TensorName.chunk( blocknum , dim)

$$分割的结果为tuple$$

A=[2,3,4,4] 两张3通道4×4图片

A=[2,3,4,4] ，A.chunk(2,dim=0)

A=[2,3,4,4] ，A.chunk(2,dim=1)

A=[2,3,4,4] ，A.chunk(2,dim=2)

A=[2,3,4,4] ，A.chunk(2,dim=3)