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

Question

我有两个对象张悔改载体： （0110010）， （0111011） 我想在两者之间进行比较，并创建一个新的张量向量： （0110010） 在循环中迭代它们非常缓慢，我知道在Matlab中有一个解决方案，所以我假设还有一个用于张量的解决方案。

Answer 1

要对仅包含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