There is torch.split 90 which is the same for the the case where you give it a single int. If you need to split into non-equal-sized arrays, you’ll need to use a loop and indexing or calls to narrow.
x = torch.arange(0, 9)
x.split(3)
(
  0
  1
  2
 [torch.FloatTensor of size 3],
  3
  4
  5
 [torch.FloatTensor of size 3],
  6
  7
  8
 [torch.FloatTensor of size 3])