0
点赞
收藏
分享

微信扫一扫

pytorch :chunk

猎书客er 2022-04-15 阅读 52
python
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 条评论