triton.language.reshape

triton.language.reshape(input, shape)

Returns a tensor with the same number of elements as input but with the provided shape.

Parameters:

• input – The input tensor.

• shape (Tuple[int]) – The new shape.