lua - 如何检查两个 Torch 张量或矩阵是否相等？

Question

我需要一个 Torch 命令来检查两个张量是否具有相同的内容，如果它们具有相同的内容，则返回 TRUE。

例如:

local tens_a = torch.Tensor({9,8,7,6});
local tens_b = torch.Tensor({9,8,7,6});

if (tens_a EQUIVALENCE_COMMAND tens_b) then ... end

我应该在这个脚本中使用什么而不是 EQUIVALENCE_COMMAND ?

我只是尝试使用 ==但它不起作用。

Answer 1

torch.eq(a, b)

eq() 实现 ==运算符比较 a 中的每个元素与 b (如果 b 是一个值)或 a 中的每个元素及其在 b 中的相应元素(如果 b 是张量)。

来自@deltheil 的替代方案:

torch.all(tens_a.eq(tens_b))