如何计算两个火炬张量中相等元素的数量也等于特定值
How to get count number of equal elements in two torch tensors that also equal a specific value
我正在使用 pytorch 并尝试计算 2 个火炬张量中相等元素的数量，这些元素也等于特定值。
即张量a=[0,1,2,0,1,2]且张量b=[0,2,1,0,2,1]
我想要 return:
- 3 当我检查 a、b 中有多少元素相等并且也等于 0 时。sum(a == b and a == 0 and b == 0) = 2,
- 0 当我检查 a、b 中有多少元素相等并且也等于 1 或 2 时。sum(a == b and a == 1,2 and b == 1,2) = 0。
谢谢！
一个直接的方法是：
torch.sum((a==b) * (a==val))
如果 val 是您要查找的值。
我正在使用 pytorch 并尝试计算 2 个火炬张量中相等元素的数量，这些元素也等于特定值。
即张量a=[0,1,2,0,1,2]且张量b=[0,2,1,0,2,1]
我想要 return:
- 3 当我检查 a、b 中有多少元素相等并且也等于 0 时。sum(a == b and a == 0 and b == 0) = 2,
- 0 当我检查 a、b 中有多少元素相等并且也等于 1 或 2 时。sum(a == b and a == 1,2 and b == 1,2) = 0。
谢谢！
一个直接的方法是：
torch.sum((a==b) * (a==val))
如果 val 是您要查找的值。