【文章推荐】chunk方法可以对张量分块,返回一个张量列表: torch.chunk tensor,chunks,dim ... pytorch中torch.cat(),torch.chunk(),torch.split()函数的使用方法.
確定! 回上一頁