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.