如何用索引列表提取Tensor中元素
收藏
在numpy中我们可以这样从数组中提取元素:
import numpy as np
arr = np.random.random((3, 4))
print(arr[:, [0 , 2]])
在paddle中有没有类似的方法,提取Tensor对象中给定索引列表中的元素呢?
前提:仅使用飞桨提供的接口。不要先转numpy处理,然后再转Tensor。
0
收藏
请登录后评论
照样print~~
再说,直接tensor.numpy()转也很方便
gather_nd试试
像numpy一样直接切片操作就可以
框架重载了很多常用的操作,像numpy的就当numpy那么用,加减乘除符号也重载了。按直觉试试,往往有惊喜~~
tensor还支持js那种Tensor.a().b().c()操作呢