这个屌丝很懒，什么也没留下！

Pytorch学习-张量的合并和分割_torch.split所有分片合并

作者：IT小白 | 2024-02-23 05:30:19

torch.split所有分片合并

可以用torch.cat方法和torch.stack方法将多个张量合并，可以用torch.split方法把一个张量分割成多个张量。

torch.cat和torch.stack有略微的区别，torch.cat是连接，不会增加维度，而torch.stack是堆叠，会增加维度。

a = torch.tensor([[1, 2],[3,4]])
b = torch.tensor([[5,6],[7,8]])
c = torch.tensor1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/IT小白/article/detail/134025?site