torch.chunk:用来将tensor分成很多个块,简而言之我理解的就是切分吧,可以在不同维度上切分。torch.chunk(tensor,chunk数,维度)代码示例:import ...
確定! 回上一頁