Wie können Elemente eines Tensors auf 4 Dezimalstellen auf der GPU in Torch begrenzt/abgerundet werden?

Question

Ich frage mich, wie kann ich die Genauigkeit der Tensoren auf GPU auf eine gewünschte Dezimalstellen abgeschnitten? Ich möchte die Elemente in einem Tensor auf 4 Dezimalstellen beschränken, aber es scheint, dass es keine eingebaute Funktion in Torch gibt, um dies effizient und effektiv zu machen. Weiß jemand wie man das machen kann?

Answer 1

Es ist immer noch nicht geradlinig die Zahlen in einem Tensor abzurunden, sondern eine einfache Lösung (die Art dient der für meine Zwecke) ist nur den Tensor Halb Präzision wie folgt zu konvertieren:

require 'cutorch' 

temp = torch.rand(3, 3) 
temp = temp:cuda() 
temp = temp:cudaHalf()