PyTorch常用张量切割和拼接方法(torch.chunk、torch.split、torch.cat和torch.stack用法详解)
1、torch.chunk首先看官方文档的解释:功能:尝试将张量拆分为指定数量的数据块,每个数据块都是输入张量的一个视图。例子:>>>torch.arange(5).chunk(3)(tensor([0,1]),tensor([2,3]),tensor([4]))>>>torch.arange(6).chunk(3)(tensor([0,1]),tensor([2,3]),tensor([4,5]