Come verificare se due tensori o matrici toriche sono uguali?

Question

Ho bisogno di un comando Torch che controlli se due tensori hanno lo stesso contenuto e restituisce VERO se hanno lo stesso contenuto.

Ad esempio:

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 

Cosa devo usare in questo script, invece di EQUIVALENCE_COMMAND?

Ho provato semplicemente con == ma non funziona.

Answer 1

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

torch.eq(a, b) 

Utensili == operatore confrontando ciascun elemento in una con b (se b è un numero) o ciascun elemento di un elemento con corrispondente b.

--update

da @deltheil

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

o ancora più semplice

torch.all(tens_a:eq(tens_b)) 

Answer 2

Prova questo se si vuole ignorare piccole differenze di precisione che sono comuni per i galleggianti

torch.all(torch.lt(torch.abs(torch.add(tens_a, -tens_b)), 1e-12))