2015-11-23 3 views
1

У меня есть два тензорных объектов каясь векторов: (0110010), (0111011) Я хотел бы сравнить между ними и создать новый тензор вектор: (0110010) Итерация по ним в петле очень медленная, я знаю, что в Matlab есть решение, поэтому я предполагаю, что для тензоров тоже есть.torch.Tensor манипуляции - Сравнение двух векторов

ответ

2

Чтобы выполнить логику и операцию для тензоров, содержащих только 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 

Для сравнения двух тензоров поэлементно вы можете использовать: эк():

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 
Смежные вопросы