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

2015-11-23 19:29:45
收藏：0
阅读：161
评论：1

我有两个张量对象代表向量：

(0110010),
(0111011)

我想比较这两个，并创建一个新的张量向量：

(0110010)

在循环中迭代它们速度非常慢，我知道在 Matlab 中有一个解决方案，所以我认为在张量中也有一个解决方案。

点赞
用户387870
用户387870

对于只包含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
2015-11-24 09:19:13

作者:

用户4364020

用户4364020
推荐文章