Function tensor3d

  • tensor3d(
        values: TensorLike3D,
        shape?: [number, number, number],
        dtype?: keyof DataTypeMap,
    ): Tensor3D

  • Creates rank-3 tf.Tensor with the provided values, shape and dtype.

    The same functionality can be achieved with tf.tensor, but in general we recommend using tf.tensor3d as it makes the code more readable.

    // Pass a nested array.
    tf.tensor3d([[[1], [2]], [[3], [4]]]).print();

    // Pass a flat array and specify a shape.
    tf.tensor3d([1, 2, 3, 4], [2, 2, 1]).print();

    Parameters

    • values: TensorLike3D

      The values of the tensor. Can be nested array of numbers, or a flat array, or a TypedArray.

    • Optionalshape: [number, number, number]

      The shape of the tensor. If not provided, it is inferred from values.

    • Optionaldtype: keyof DataTypeMap

      The data type.

    Returns Tensor3D

    Doc

Settings

Member Visibility