Function tensor4d

  • tensor4d(
        values: TensorLike4D,
        shape?: [number, number, number, number],
        dtype?: keyof DataTypeMap,
    ): Tensor4D

  • Creates rank-4 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.tensor4d as it makes the code more readable.

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

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

    Parameters

    • values: TensorLike4D

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

    • Optionalshape: [number, number, number, number]

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

    • Optionaldtype: keyof DataTypeMap

      The data type.

    Returns Tensor4D

    Doc

Settings

Member Visibility