如何在火炬中的GPU上將張量的元素限制爲四個小數位？

Question

我想知道，如何將GPU上張量的精度截斷到所需的小數位？我想將張量中的元素限制在4個小數位，但似乎Torch中沒有內置函數來有效地完成此操作。有誰知道如何做到這一點？

Answer 1

它仍然不是直截了當圓的數字在張量，但一個簡單的解決方案（哪一種的目的是爲了我的目的）是剛剛張量轉換爲半精度如下：

require 'cutorch' 

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