Python中文
首页
教程
问答
标签
搜索
登录
注册
Numpy/Torch:使用维度上的索引选择元素
回答此问题可获得
20
贡献值,回答如果被采纳可获得
50
分。
<div> </div> <p>给定如下所示的数组:</p> <pre><code>np.arange(12).reshape(4,3) Out[119]: array([[ 0, 1, 2], [ 3, 4, 5], [ 6, 7, 8], [ 9, 10, 11]]) </code></pre> <p>我想使用索引列表<code>[0, 2, 1, 2]</code>从每一行中选择一个元素,以创建<code>[0, 5, 7, 11]</code>的<code>4x1</code>数组</p> <p>有没有简单的方法来建立索引。我能看到的最接近的方法是<code>pytorch</code>中的<code>gather</code>方法</p>
0 条评论
分类:
Python问答
请先
登录
后评论
默认排序
时间排序
1 个回答
匿名
1天前
擅长:python、mysql、java
<pre><code>arr[[0,1,2,3], [0,2,1,2]] </code></pre> <p>或者,如果您更喜欢<code>np.arange(4)</code>作为第一个索引数组</p>
请先
登录
后评论
针对此问题:
更多的回答
关注
89
关注
收藏
1
收藏,
216
浏览
网友 提问于 2天前
相关Python问题
Django:。是不是“超级用户”字段不起作用
3 回答
Django:'DeleteQuery'对象没有属性'add'
6 回答
Django:'ModelForm'对象没有属性
7 回答
Django:'python manage.py runserver'返回'TypeError:'WindowsPath'类型的对象没有len()
4 回答
Django:'Python管理.pysyncdb'不创建我的架构表
8 回答
Django:'Python管理.py迁移“耗时数小时(和其他奇怪的行为)
7 回答
Django:'readonly'属性在我的ModelForm上不起作用
7 回答
Django:'RegisterEmployeeView'对象没有属性'object'
9 回答
Django:'str'对象没有属性'get'
9 回答
Django:'创建' 不能被指定为Order模型表单中的值,因为它是一个不可编辑的字段
9 回答
Django:“'QuerySet'类型的对象不是JSON可序列化的”
6 回答
Django:“'utf8'编解码器无法解码位置19983中的字节0xe9:无效的连续字节”,加载临时文件时
5 回答
Django:“<…>”需要有一个字段“id”的值,然后才能使用这个manytomy关系
5 回答
Django:“AnonymousUser”对象没有“get_full_name”属性
4 回答
Django:“ascii”编解码器无法解码位置1035中的字节0xc3:序号不在范围内(128)
10 回答
Django:“BaseTable”对象不支持索引
3 回答
Django:“collections.OrderedDict”对象不可调用
2 回答
Django:“Country”对象没有属性“all”
1 回答
Django:“Data”对象没有属性“save”
6 回答
Django:“datetime”类型的对象不是JSON serializab
9 回答