Wie um zu überprüfen, ob zwei Brenner Tensoren oder Matrizen sind gleich?
Brauche ich eine Fackel-Befehl, der prüft, ob zwei Tensoren haben die gleichen Inhalte, und gibt TRUE zurück, wenn Sie den gleichen Inhalt haben.
Beispiel:
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
Was sollte ich in diesem script statt EQUIVALENCE_COMMAND
?
Habe ich versucht einfach mit ==
aber es funktioniert nicht.
- Zu ermöglichen, für floating-point-Unterschiede, siehe Überprüfen Sie, ob PyTorch Tensoren sind gleich innerhalb von epsilon.
Du musst angemeldet sein, um einen Kommentar abzugeben.
https://github.com/torch/torch7/blob/master/doc/maths.md#torcheqa-b
Implementiert = = - operator vergleicht jedes element in a mit b (wenn b eine Zahl ist) oder jedes element in a mit dem entsprechenden element in b.
--UPDATE
vom @deltheil
oder noch einfacher
torch.all
, d.h.torch.all(torch.eq(tens_a, tens_b))
oder noch einfachertorch.all(tens_a:eq(tens_b))
.torch.all
.torch.eq(tens_a, tens_b).all()
t.all()
am Ende:
syntax?Versuchen Sie dies, wenn Sie Sie ignorieren wollen kleine Präzisions-Unterschiede, die Häufig für Schwimmer
Diese unter Lösung bei mir funktioniert:
Vom die Dokumentation: