Ejemplos de native_roll en Python

Ejemplo n.º 1

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

Ejemplo n.º 2

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