I believe I should be using torch.split() 915

x = torch.rand((3,4,8))
slices = torch.split(x, 4, 2)


print(len(slices))
> 2


print(slices[0].size())
> torch.Size([3, 4, 4])

This will nicely return tuples of 3D tensors.
PS. Just posting as it might be a help to one in need.