PyTorch常用函数总结

将一个tensor分到多个GPU上:torch.cuda.comm.scatter