假设我有一个看起来像这样的二维张量:
[[44, 50, 1, 32],
.
.
.
[7, 13, 90, 83]]
以及我想要选择的行索引列表,看起来像这样 [0, 34, 100, ..., 745]
。我如何遍历并创建一个新的张量,该张量仅包含其索引包含在数组中的行?
最佳答案
你可以像使用 numpy 一样选择
import torch
x = torch.Tensor([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 8, 7, 6],
[5, 4, 2, 1]])
indices = [0, 3]
print(x[indices])
# tensor([[1., 2., 3., 4.],
# [5., 4., 2., 1.]])
https://stackoverflow.com/questions/60032073/
相关文章:
awk - 将 'u' 转换为 't' 而无需更改 FASTA 中的 header
javascript - 数组中的对象 数组中的对象 数组中的对象 JavaScript - 搜索
javascript - 使用 axios 使用 get 方法发送嵌套的 json 数据
javascript - Angular 类型 'split' 上不存在属性 'ArrayBuffe
docker - 我想在正在运行的 docker 容器中使用 docker sdk,并想在正在运行的
kubernetes - 集群上的事件与 kustomize 之间的区别