[7, 8, 9]])
x = torch.from_numpy(x)
print(‘before roll’, x)
x = torch.roll(x, 1)
print(‘after roll’, x)
3.2 例2 — shifts=1 & dims=0
x = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
x = torch.from_numpy(x)
print(‘before roll’, x)
x = torch.roll(x, 1, dims=0)
print(‘after roll’, x)