pytorch tensor操作问题. 如图所示. 求操作方法..

zPeregrine-avatar

zPeregrine

2022-02-01T07:12:46+00:00

[img]https://img.nga.178.com/attachments/mon_202202/03/-7Q2p-25tjKmT1kSgs-7b.jpg[/img]

我有两个矩阵, 我想判断第二个矩阵中的每一个元素, 是否存在于第一个矩阵的每一行中. 该如何操作?

图上第三行的操作没有画出来 . 因为没地方画了. 第三行的计算结果应该是false.
Imbeast-avatar

Imbeast

any(a==b)
zPeregrine-avatar

zPeregrine

[quote][pid=586282396,30538373,1]Reply[/pid] Post by [uid=60199305]云卷花开[/uid] (2022-02-03 15:33):

any(a==b)[/quote]确实可以, 感谢大佬
Imbeast-avatar

Imbeast

[quote][pid=586282760,30538373,1]Reply[/pid] Post by [uid=63478635]jueduiwushi[/uid] (2022-02-03 15:35):

确实可以, 感谢大佬[/quote]那就torch.eq之类的方法吧,我记不太清楚了
Kurai-avatar

Kurai

次奥就真泥潭人均机器学习?
memoir-avatar

memoir

(a==b).long().sum(0).bool()

tensor([ True, True, False])