torch.chunk()函数

 

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

示例：