Wie Pytorch Zeilen Normalisierung für jede matrix in ein 3D-Tensor(Variable)?

Wenn ich eine 3D-Tensor (Variable) mit der Größe [a,b,c].
betrachten Sie es als b*c-matrix, und ich hoffe, dass all diese matrix bekam Zeile normalisiert.

InformationsquelleAutor Qinqing Liu | 2017-11-21
  1. 8

    Können Sie die normalize Funktion.

    import torch.nn.functional as f
f.normalize(input, p=2, dim=2)

    Den dim=2 argument gibt an, entlang welcher dimension zu normalisieren (teilen, jede Zeile Vektor durch seine p-norm.

    • Ich habe versucht, aber die Zeile Summe ist nicht 1. Danke aber. Ich habe auch versucht renorm-Funktion, es funktioniert auch nicht.
    • Tun Sie etwas dagegen Aktualisierung der Frage mit Angaben mindestens zu arbeiten Beispiel was nicht, geben die Ausgabe? Ich werde in einer besseren position zu ändern, dann.
    • wenn Sie möchten, die Zeile Summe gleich 1 (L1-norm) können Sie einstellen p=1 diese Antwort legt die Summe der Quadrate von 1 (L2-norm).
    InformationsquelleAutor activatedgeek
  2. 1

    Folgende arbeiten sollten.

    import torch
import torch.nn.functional as f

a, b, c = 10, 20, 30

t = torch.rand(a, b, c)
g = f.normalize(t.view(t.size(0), t.size(1) * t.size(2)), p=1, dim=1)
print(g.sum(1)) # it confirms the normalization

g = g.view(*t.size())
print(g) # get the normalized output vector of shape axbxc
    InformationsquelleAutor Wasi Ahmad
Schreibe einen Kommentar