torch.chunk()函数

 

将tensor按dim方向分割成chunks个tensor块，返回的是一个元组。

示例：

 

posted @ 2020-12-01 22:41  Picassooo  阅读(2928)  评论(0)  编辑  收藏  举报