pytorch中torch.chunk()方法

对tensor张量在指定维度上进行分块，返回的是一个张量元组。

详见：https://www.cnblogs.com/jiangkejie/p/10309766.html