torch.cat()拼接两个矩阵

最新推荐文章于 2025-03-31 17:34:39 发布

原创

最新推荐文章于 2025-03-31 17:34:39 发布 · 4.1k 阅读

2 ·

CC 4.0 BY-SA版权

本文详细介绍了如何使用PyTorch进行二维和三维矩阵的操作，包括矩阵的创建、拼接等核心过程。通过具体示例展示了按行和按列拼接矩阵的方法，以及在三维矩阵中如何进行维度调整和拼接，为读者提供了丰富的实践指导。

二维矩阵

import torch

if __name__ == '__main__':
    a = torch.tensor([[1,1,1],
                      [2,2,2],
                      [3,3,3]],dtype=torch.float)

    b = torch.tensor([[1,0,1],
                      [2,0,2],
                      [3,0,3]],dtype=torch.float)
    c = torch.cat( (a, b), 0)#按行拼接
    d = torch.cat( (a, b), 1