The array in my example above is created by indexing my network’s output, array = output[0,:,i,j,k]:
>>> output.shape
torch.Size([1, 151, 137, 199, 178])
>>> output.stride()
(732774914, 4852814, 35422, 178, 1)

>>> array.shape
torch.Size([151])
>>> array.stride()
(1,)