首页 > 编程语言 > 详细

libtorch 数组索引和张量操作 与pytorch比较

时间:2021-09-02 05:46:53      阅读:24      评论:0      收藏:0      [点我收藏+]

数组索引

  1. libtorch

    torch::Tensor idx=torch::zeros({270,360});
    torch::Tensor t=torch::zeros({255});
    //idx里的值作为t的索引
    torch::Tensor out=t.index({idx})

  2. python
    out=t[idx]

张量索引

1.python
t[i,j]
2.libtorch
t.index({i,j})

libtorch 数组索引和张量操作 与pytorch比较

原文:https://www.cnblogs.com/ggs403/p/15213289.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!