ttnn.roll

ttnn.roll = FastOperation(python_fully_qualified_name='ttnn.roll', function=<ttnn._ttnn.operations.data_movement.roll_t object>, preprocess_golden_function_inputs=None, golden_function=None, postprocess_golden_function_outputs=None, is_cpp_operation=True, is_experimental=False)

Performs circular shifting of elements along the specified dimension(s).

Parameters:

• input_tensor – A tensor whose elements will be rolled.

• shifts – The number of places by which elements are shifted. Can be an integer or a list of integers (one per dimension).

• dim – The dimension(s) along which to roll. If shifts is a list, then dim must be a list of the same length as shifts.