torch.Tensor 操作 - 比较两个向量

torch.Tensor manipulation - Comparing two vectors

我有两个张量对象忏悔向量: (0110010), (0111011) 我想比较两者并创建一个新的张量向量: (0110010) 在循环中迭代它们非常慢,我知道在 Matlab 中有一个解决方案,所以我假设张量也有一个。

要对仅包含 1 和 0 元素的张量执行逻辑与运算,您可以使用 :cmul() 成员函数(逐元素乘法)。

th> torch.Tensor({0,1,1,0,0,1,0}):cmul(torch.Tensor({0,1,1,1,0,1,1}))
 0
 1
 1
 0
 0
 1
 0

要按元素比较两个张量,您可以使用 :eq():

th> torch.Tensor({0,1,1,0,0,1,0}):eq(torch.Tensor({0,1,1,1,0,1,1}))
 1
 1
 1
 0
 1
 1
 0