Equivalent function like `numpy.diff` in pytorch?

Would this work for you?

x = torch.arange(4*5*6).view(4, 5, 6)
idx = (x == 82).nonzero()
1 Like