def roll(tensor, shifts, **kwargs): if isinstance(shifts, FrameworkTensor): shifts = int(shifts.item()) return torch.native_roll(tensor, shifts, **kwargs)
def roll(tensor, shifts, **kwargs): int_shifts = int(shifts.item()) return torch.native_roll(tensor, int_shifts, **kwargs)