2016-05-24 8 views
0

ich für einige effiziente Weg suchen (oder eine Kombination von Möglichkeiten) die Anzahl der Instanzen (dh erhalten Frequenz) eines bestimmten Elements in einer allgemeinen zu zählen torch.Tensor.torch7: Erhalten der Frequenz eines Elements in einem Tensor

Mit effizient, ich meine etwas anderes als die triviale iterative Methode, um jedes Element des gegebenen Tensors und etwas, die Vorteil der Parallelität von den GPUs angeboten werden können.

Ich habe in der Dokumentation sah tensor.md zur Verfügung gestellt und maths.md aber war nicht in der Lage, etwas zu finden, die mir helfen könnte.

+1

'T: eq (x): Summe()' –

Antwort

1

Nur Alex Kommentar als Antwort formulieren. Ich fand es nützlich für meine eigene Arbeit. Beispiel:

t = torch.Tensor({1, 2, 3, 1, 4, 2, 2, 2, 3, 0}) 
frequency = t:eq(2):sum() -- frequency of 2 in t