欢迎大家交换友链，可在https://www.stubbornhuang.com/申请友情链接进行友链交换申请！
本站由于前段时间遭受到大量临时和国外邮箱注册，所以对可注册的邮箱类型进行了限制！
问题反馈可发送邮件到stubbornhuang@qq.com
计算机图形学与计算几何经典必备书单整理，下载链接可参考：https://www.stubbornhuang.com/1256/
感谢大家访问本站，希望本站的内容可以帮助到大家！
工资「喂饱肚子」，副业「养活灵魂」！
在本站开通年度VIP，无限制下载本站资源和阅读本站文章
本站会放置Google广告用于维持域名以及网站服务器费用。
如果觉得本站的内容有帮助，可以考虑打赏博主哦！

Pytorch – torch.cat函数

StubbornHuang Pytorch 发布于2022-12-08 阅读 1,972次 0次评论 0次点赞本文共553个字，阅读需要2分钟。

1 torch.cat函数

形式

torch.cat(tensors, dim=0, *, out=None)

功能

在指定的维度连接给定序列的张量，所有张量必须具有相同的形状（连接维度除外）或者为空。

参数

tensors：相同形状的张量序列，非空张量必须具有相同形状（连接维度除外）
dim：张量连接的维度

使用示例

import torch

if __name__ == '__main__':
    a = torch.randn(size=(2, 2))
    print(a)

    b = torch.randn(size=(2, 2))
    print(b)

    c = torch.cat((a, b), dim=0)
    print(c)

    d = torch.cat((a, b), dim=1)
    print(d)

输出

tensor([[ 0.8793,  0.3727],
        [-2.3334, -1.4567]])
tensor([[-1.0906, -1.2683],
        [ 0.7161, -0.5843]])
tensor([[ 0.8793,  0.3727],
        [-2.3334, -1.4567],
        [-1.0906, -1.2683],
        [ 0.7161, -0.5843]])
tensor([[ 0.8793,  0.3727, -1.0906, -1.2683],
        [-2.3334, -1.4567,  0.7161, -0.5843]])

欢迎扫码关注我的微信公众号，及时获取文章更新