pytorch :chunk

阅读 55

2022-04-15

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)

相关推荐

精彩评论(0)

0 0 举报