python - Pytorch - 如何沿轴索引具有不同大小的张量

Question

我有一个T形状的张量，(8, 8, 4, 4)我有 2 个向量，x用于y索引这个张量的最后 2 个维度。

例如：

x = tensor([1, 2, 3, 1, 2, 3, 1, 2])
y = tensor([2, 3, 2, 3, 2, 3, 2, 3])

我想将张量索引处的元素设置为-1. 结果预计与此 for 循环相同，但仅适用于没有任何循环的 pytorch。

for i in range(8):
    for j in range(8):
        T[i, j, :x[i], :y[j]] = -1

这可以在没有任何循环的情况下完成吗？如果是这样，怎么做？谢谢。

标签： pythonindexingpytorch