torch.chunk()函数

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