Getting Started
Python API
reshape()
Triton MLIR Dialects
Programming Guide
Returns a tensor with the same number of elements as input but with the provided shape.
input – The input tensor.
shape (Tuple[int]) – The new shape.