Tensor对象的索引
Tensor对象的索引
Tensor 默认是支持[]操作符的，因此可以使用这样的方式来获取元素：
auto foo = torch::randn({1, 2, 3, 4});
float value = foo[0][1][2][2];另一种方式是用Tensor对象的index函数，它的优势是支持slice。 对于单个元素，可以类似Pytorch中，直接用index({i, j, k})的方式来索引：
auto foo = torch::randn({1, 2, 3, 4});
float value = foo.index({0, 1, 2, 2});那么python中很常用的slice呢？例如foo[..., :2, 1:, :-1]，该怎么在Libtorch中表示？ 这里需要用到torch::indexing::Slice 对象，来实现Python中的Slice，看看下面的例子你就明白了：
using namespace torch::indexing;
auto foo = torch::randn({1, 2, 3, 4});
// 等效于Python中的foo[:, 0:1, 2:, :-1]
auto bar = foo.index({Slice(), Slice(0, 1), Slice(2, None), Slice(None, -1)});应该是能满足Python中slice同样的使用场景。
网站声明：如果转载，请联系本站管理员。否则一切后果自行承担。
赞同 0
评论 0 条
- 上周热门
- 银河麒麟添加网络打印机时，出现“client-error-not-possible”错误提示 1487
- 银河麒麟打印带有图像的文档时出错 1405
- 银河麒麟添加打印机时，出现“server-error-internal-error” 1194
- 统信操作系统各版本介绍 1116
- 统信桌面专业版【如何查询系统安装时间】 1114
- 统信桌面专业版【全盘安装UOS系统】介绍 1069
- 麒麟系统也能完整体验微信啦！ 1026
- 统信【启动盘制作工具】使用介绍 672
- 统信桌面专业版【一个U盘做多个系统启动盘】的方法 616
- 信刻全自动档案蓝光光盘检测一体机 526
- 本周热议
- 我的信创开放社区兼职赚钱历程 40
- 今天你签到了吗？ 27
- 信创开放社区邀请他人注册的具体步骤如下 15
- 如何玩转信创开放社区—从小白进阶到专家 15
- 方德桌面操作系统 14
- 我有15积分有什么用? 13
- 用抖音玩法闯信创开放社区——用平台宣传企业产品服务 13
- 如何让你先人一步获得悬赏问题信息？（创作者必看） 12
- 2024中国信创产业发展大会暨中国信息科技创新与应用博览会 9
- 中央国家机关政府采购中心：应当将CPU、操作系统符合安全可靠测评要求纳入采购需求 8
热门标签更多
