gpt4 book ai didi

python - 如何检查某个数字是否在 Pytorch 张量中?

转载 作者:行者123 更新时间:2023-12-03 22:00:12 25 4
gpt4 key购买 nike

对于 Pytorch 张量 A :

A = tensor([1,0,0],
[0,0,0])


有什么方法可以检查数字 1 是否是张量的元素 A ?
比如是否有一个返回 True 的 pytorch 函数是 1 是 A 的一个元素, 如果 1 不是 A 的元素,则返回 False ?

谢谢,

最佳答案

torch.Tensor工具__contains__ .所以,你可以使用:

1 in A

这将返回 True如果元素 1A , 和 False除此以外。

关于python - 如何检查某个数字是否在 Pytorch 张量中?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60153144/

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