Class Ones<T extends TType>

java.lang.Object
org.tensorflow.framework.initializers.BaseInitializer<T>
org.tensorflow.framework.initializers.Ones<T>
Type Parameters:
T - The TType for the call operation
All Implemented Interfaces:
Initializer<T>

public class Ones<T extends TType> extends BaseInitializer<T>
Initializer that generates tensors initialized to 1.

Examples:

     Ones<TFloat32> initializer =
             new org.tensorflow.framework.initializers.Ones<>(tf);
     Operand<TFloat32> values =
             initializer.call(Ops tf, tf.constant(Shape.of(2,2)), TFloat32.class);
  • Constructor Details

    • Ones

      public Ones()
      Creates an Initializer that sets all values to one.

      Examples:

           Ones<TFloat32> initializer =
                   new org.tensorflow.framework.initializers.Ones<>(tf);
           Operand<TFloat32> values =
                   initializer.call(Ops tf, tf.constant(Shape.of(2,2)), TFloat32.class);
      
  • Method Details

    • call

      public Operand<T> call(Ops tf, Operand<TInt64> dims, Class<T> type)
      Generates the operation used to perform the initialization.
      Parameters:
      tf - the TensorFlow Ops
      dims - the shape dimensions
      type - the type of tensor
      Returns:
      An operand for the initialization.