PyTorch常用函数总结

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

posted @ 2019-07-20 09:27  洗盏更酌  Views(225)  Comments(0)  Edit  收藏  举报