torch.chunk()函数

 

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

示例:

 

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