TorchのGPUでテンソルの要素を小数点以下4桁に制限する方法は？

Question

GPUのテンソル精度を小数点以下の桁数に切り捨てるにはどうすればよいですか？私はテンソルの要素を小数点以下4桁に制限したいが、効率的かつ効果的にこれを行うための組み込み関数はTorchにはないようだ。誰かがこれをどうやってできるか知っていますか？

Answer 1

それはテンソルで数値を丸めることはまだストレートフォワードではないですが（一種の私の目的のためになります）簡単な解決策は、単に次のように半精度にテンソルを変換することです：

require 'cutorch' 

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