Проверьте равенство любых верхних k записей в строках тензора A и argmax в строках тензора B

#python #pytorch #tensor Вопрос: Новое в тензорах/pytorch. У меня есть два 2d тензора, A и B. A содержит поплавки, представляющие вероятность, присвоенную определенному индексу. B содержит один горячий двоичный вектор…