• Reshapes a tf.Tensor to a given shape.

    Given an input tensor, returns a new tensor with the same values as the input tensor with shape shape.

    If one component of shape is the special value -1, the size of that dimension is computed so that the total size remains constant. In particular, a shape of [-1] flattens into 1-D. At most one component of shape can be -1.

    If shape is 1-D or higher, then the operation returns a tensor with shape shape filled with the values of tensor. In this case, the number of elements implied by shape must be the same as the number of elements in tensor.

    const x = tf.tensor1d([1, 2, 3, 4]);
    x.reshape([2, 2]).print();

    Type Parameters

    Parameters

    • x: TensorLike | Tensor<Rank>

      The input tensor to be reshaped.

    • shape: ShapeMap[R]

      An array of integers defining the output tensor shape.

    Returns Tensor<R>