torch.cat((A,B),dim=-1)

最新推荐文章于 2024-04-24 15:52:45 发布

茶茶小甜豆

最新推荐文章于 2024-04-24 15:52:45 发布

阅读量5.3k

点赞数 3

文章标签：深度学习人工智能 python

本文链接：https://blog.csdn.net/zyd_1129/article/details/127649231

版权

>>> import torch
>>> x = torch.randn(2,3)
>>> x
tensor([[-0.8168, -1.9389,  0.0781],
        [ 0.3570, -1.3199, -0.1600]])
>>> torch.cat((x,x),0)
tensor([[-0.8168, -1.9389,  0.0781],
        [ 0.3570, -1.3199, -0.1600],
        [-0.8168, -1.9389,  0.0781],
        [ 0.3570, -1.3199, -0.1600]])
>>> torch.cat((x,x),-1)
tensor([[-0.8168, -1.9389,  0.0781, -0.8168, -1.9389,  0.0781],
        [ 0.3570, -1.3199, -0.1600,  0.3570, -1.3199, -0.1600]])

dim=-1是两块直接拼接