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