首页 Paddle框架 帖子详情
如何用索引列表提取Tensor中元素
收藏
快速回复
Paddle框架 问答深度学习炼丹技巧 1161 6
如何用索引列表提取Tensor中元素
收藏
快速回复
Paddle框架 问答深度学习炼丹技巧 1161 6

在numpy中我们可以这样从数组中提取元素:

import numpy as np

arr = np.random.random((3, 4))

print(arr[:, [0 , 2]])

在paddle中有没有类似的方法,提取Tensor对象中给定索引列表中的元素呢?

前提:仅使用飞桨提供的接口。不要先转numpy处理,然后再转Tensor。

0
收藏
回复
全部评论(6)
时间顺序
AIStudio810258
#2 回复于2021-01

照样print~~

0
回复
AIStudio810258
#3 回复于2021-01

再说,直接tensor.numpy()转也很方便

0
回复
韩憨
#4 回复于2021-06

gather_nd试试

0
回复
FutureSI
#5 回复于2021-06

像numpy一样直接切片操作就可以

0
回复
FutureSI
#6 回复于2021-06

框架重载了很多常用的操作,像numpy的就当numpy那么用,加减乘除符号也重载了。按直觉试试,往往有惊喜~~

0
回复
FutureSI
#7 回复于2021-06

tensor还支持js那种Tensor.a().b().c()操作呢

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户