Torch.chunk() method in pytorch

Block the tensor tensor in the specified dimension, and return a tensor tuple.

For details, please see: https://www.cnblogs.com/jiangkejie/p/10309766.html