triton.language.view

triton.language.view(input, shape)

Returns a tensor with the same elements as input but a different shape. The order of the elements may not be preserved.

Parameters:
  • input – The input tensor.

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