export function makeInitializer()

in rfcs/20210731-tfjs-named-tensors/gtensor.ts [403:415]


export function makeInitializer(config: InitializerConfig) {
  if (config.tuncNormal) {
    return tf.initializers.truncatedNormal(config.tuncNormal);
  } else if (config.zeros) {
    return tf.initializers.zeros();
  } else if (config.ones) {
    return tf.initializers.ones();
  } else if (config.constant) {
    return tf.initializers.constant(config.constant);
  }

  throw new ValueError('need to specify an initalizer config');
}