triton.language.reshape

triton.language.reshape(input, shape)

Tries to reshape the given tensor to a new shape.

Parameters
  • input – The input tensor.

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