2つのTensor型配列を走査し一致している要素数を取得する

参考記事：stackoverflow.com
2つのTensor型配列AとBについて、まず "A == B" によって、要素が一致しているインデックスはTrue、一致していないインデックスはFalseとなっている配列を取得する。次に、"(A == B).sum()" または "torch.sum(A == B)" によってTrueの個数すなわち一致した要素数をTensor型で取得する。int()を使えばint型にキャストできる。

>>> import torch
>>> a = torch.tensor([1, 3, 2, 0, 1, 2, 1, 0, 0, 1])
>>> b = torch.tensor([1, 0, 0, 0, 1, 3, 1, 1, 2, 0])
>>> a == b
tensor([ True, False, False,  True,  True, False,  True, False, False, False])
>>> (a == b).sum()
tensor(4)
>>> torch.sum(a == b)
tensor(4)
>>> int((a == b).sum())
4