gpt4 book ai didi

python - Pytorch Tensor如何获取特定值的索引

转载 作者:太空狗 更新时间:2023-10-29 17:04:03 26 4
gpt4 key购买 nike

使用 python 列表,我们可以:

a = [1, 2, 3]
assert a.index(2) == 1

pytorch 张量如何直接找到 .index()

最佳答案

我认为没有从 list.index() 到 pytorch 函数的直接转换。但是,您可以使用 tensor==number 然后使用 nonzero() 函数获得类似的结果。例如:

t = torch.Tensor([1, 2, 3])
print ((t == 2).nonzero(as_tuple=True)[0])

这段代码返回

1

[torch.LongTensor of size 1x1]

关于python - Pytorch Tensor如何获取特定值的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47863001/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com