Thought as much.  I can do that but I assume this is going to bring the tensor from the GPU to the CPU and then will have to put it back.  That doesn’t seem very efficient.  I’ll see if I can implement the same thing.  I want the velocity from each sample but was kinda lazy since this accomplishes almost the same (well, not really np.diff) and preserves the shape for my purposes.
I’ll see if I can do something with it.