Function ones

  • ones<R>(shape, dtype?): Tensor<R>

  • Creates a tf.Tensor with all elements set to 1.

    tf.ones([2, 2]).print();

    Type Parameters

    • R extends Rank

    Parameters

    • shape: ShapeMap[R]

      An array of integers defining the output tensor shape.

    • Optional dtype: keyof DataTypeMap

      The type of an element in the resulting tensor. Defaults to 'float'.

    Returns Tensor<R>

    Doc

Settings

Member Visibility

Theme

Generated using TypeDoc