我需要一个 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
?我只是尝试使用
==
但它不起作用。
最佳答案
torch.eq(a, b)
eq()
实现 ==
运算符比较 a
中的每个元素与 b
(如果 b 是一个值)或 a
中的每个元素及其在 b
中的相应元素(如果 b
是张量)。来自@deltheil 的替代方案:
torch.all(tens_a.eq(tens_b))
关于lua - 如何检查两个 Torch 张量或矩阵是否相等?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32996281/