Python wrappers
Public Python APIs implemented by this engine symbol.Circularly shift elements along one or more axes.
For each (shift, axis) pair, rotates the tensor along that axis by
shift positions; elements that fall off one end wrap around to the
other.
Parameters
aTensorImplPtrInput tensor.
shiftsvector<int64>Signed shift amount per axis.
axesvector<int>Axes to shift. Must match
shifts in length.Returns
TensorImplPtrTensor of the same shape and dtype as a.
Notes
Backward applies the inverse circular shift (negated shifts) to the
incoming gradient, which exactly undoes the forward rotation.