【问题标题】:torch.Tensor manipulation - Comparing two vectorstorch.Tensor 操作 - 比较两个向量
【发布时间】:2016-02-26 00:46:54
【问题描述】:

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

【问题讨论】:

    标签: vector lua torch


    【解决方案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 
    

    【讨论】:

      猜你喜欢
      • 2010-12-25
      • 2016-02-25
      • 1970-01-01
      • 2016-06-29
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-06-15
      • 2014-02-24
      相关资源
      最近更新 更多