Python中文
首页
教程
问答
标签
搜索
登录
注册
Pytorch张量获取具有特定值的元素的索引?
回答此问题可获得
20
贡献值,回答如果被采纳可获得
50
分。
<p>我有两个张量,张量a和张量b</p> <p>我想得到张量b中所有值的索引</p> <p>比如说</p> <pre><code>a = torch.Tensor([1,2,2,3,4,4,4,5]) </code></pre> <pre><code>b = torch.Tensor([1,2,4]) </code></pre> <p>我想要张量a中的<code>1, 2, 4</code>的索引。我可以通过以下代码来实现这一点</p> <pre class="lang-py prettyprint-override"><code>a = torch.Tensor([1,2,2,3,4,4,4,5]) b = torch.Tensor([1,2,4]) mask = torch.zeros(a.shape).type(torch.bool) print(mask) for e in b: mask = mask + (a == e) print(mask) </code></pre> <p>没有<code>for</code>我怎么做</p>
0 条评论
分类:
Python问答
请先
登录
后评论
默认排序
时间排序
1 个回答
匿名
1天前
擅长:python、mysql、java
<p>这就是你想要的吗</p> <pre class="lang-py prettyprint-override"><code>np.in1d(a.numpy(), b.numpy()) </code></pre> <p>将导致:</p> <pre><code>array([ True, True, True, False, True, True, True, False]) </code></pre>
请先
登录
后评论
针对此问题:
更多的回答
关注
89
关注
收藏
1
收藏,
216
浏览
网友 提问于 2天前
相关Python问题
Python中两个字典的交集
3 回答
python中两个字符串上的异或操作数?
6 回答
Python中两个字符串中的类似句子
5 回答
Python中两个字符串之间的Hamming距离
8 回答
python中两个字符串之间的匹配模式
4 回答
python中两个字符串之间的按位或
3 回答
python中两个字符串之间的数据(字节)切片
9 回答
python中两个字符串之间的模式
9 回答
python中两个字符串作为子字符串的区别
7 回答
Python中两个字符串元组的比较
4 回答
Python中两个字符串列表中的公共字符串
7 回答
python中两个字符串的Anagram测试
4 回答
Python中两个字符串的正则匹配
8 回答
python中两个字符串的笛卡尔乘积
6 回答
Python中两个字符串相似性的比较
8 回答
python中两个字符串语义相似度的求法
5 回答
Python中两个字符置换成固定长度的字符串,每个字符的数目相等
2 回答
Python中两个对数方程之间的插值和平滑数据
1 回答
Python中两个对象之间的And/Or运算符
1 回答
python中两个嵌套字典中相似键的和值
7 回答