TensorName.chunk( blocknum , dim)
分
割
的
结
果
为
t
u
p
l
e
分割的结果为tuple
分割的结果为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)