torch.cat torch.concatenate torch.concat学习记录

2023-12-20 15:31:25

?torch.cat, torch.concat, torch.concatenate将向量按照指定的维度进行拼接。这三个函数是同一个函数，互为彼此的别名。函数的定义函数写在torch.cat。

x = torch.randn(2, 3)
print(x)
'''
tensor([[ 0.7004, -0.0935, -0.2668],
        [ 0.7922,  0.9567,  1.4191]])
'''
'''
将三个x tensor连接在一起，连接维度为0维
'''
x_row = torch.cat((x, x, x), 0)

print(x_row)

'''
将两个x tensor连接在一起，连接维度为1维
'''
x_col = torch.cat((x, x), 1)
print(x_col)
'''
tensor([[ 0.7004, -0.0935, -0.2668],
        [ 0.7922,  0.9567,  1.4191],
        [ 0.7004, -0.0935, -0.2668],
        [ 0.7922,  0.9567,  1.4191],
        [ 0.7004, -0.0935, -0.2668],
        [ 0.7922,  0.9567,  1.4191]])
tensor([[ 0.7004, -0.0935, -0.2668,  0.7004, -0.0935, -0.2668],
        [ 0.7922,  0.9567,  1.4191,  0.7922,  0.9567,  1.4191]])
'''

将tensor进行拼接，拼接的维度根据dim设置，默认为0（行拼接）

若dim设置为-1，则拼接维度会自动选择

文章来源:https://blog.csdn.net/weixin_41147796/article/details/135102369
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。