ttnn.roll

ttnn.roll(input_tensor, shifts, dim) → None

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.