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

我需要一个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使用什么？

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

最佳答案

https://github.com/torch/torch7/blob/master/doc/maths.md#torcheqa-b

torch.eq(a, b)

实现==运算符，将a中的每个元素与b(如果b是一个数字)或a中的每个元素与b中的对应元素进行比较。
更新
来自@deltheil

torch.all(torch.eq(tens_a, tens_b))

甚至更简单

torch.all(tens_a.eq(tens_b))