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

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

    Parameters

    • values: TensorLike2D

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

    • Optionalshape: [number, number]

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

    • Optionaldtype: keyof DataTypeMap

      The data type.

    Returns Tensor2D