pytorch 一つの配列をいくつかの配列に分断したい

Question

sizeが[64,32,100,100]

のものを

[64,4,100,100]*８個

にしたいです [:]　x[:.0]などを使っていろいろやってみましたがうまくいきません
viewなどを調べてみましたが理解が足らずうまくいきません　
できる人教えてください

Answer 1

a=torch.randn(3,8,3)
a[2]*=0
print(a)
a=a.view(2,a.size(0),4,a.size(2))
でなんとかいけました