トーチ、変数をチェックする方法はCUDAかどうか？

Question

私はどの変数がCudaTensorかNormalかを識別するためにtype（）のような関数を探しています。

require('cutorch') 

x = torch.Tensor(3,3) 
x = x:cuda() 

if type(x) == 'CudaTensor' then -- What function should be used? 
    print('x is CUDA tensor') 
else 
    print('x is normal tensor') 
end 

Answer 1

使用:type()テンソルの方法：

cutorch = require('cutorch') 

x = torch.Tensor(3,3) 
x = x:cuda() 

if x:type() == 'torch.CudaTensor' then 
    print('x is CUDA tensor') 
else 
    print('x is normal tensor') 
end