torch.chunk()函数

 

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

示例：

【推广】 免费学中医，健康全家人
原文地址：https://www.cnblogs.com/picassooo/p/14071304.html