Факел, как проверить переменную CUDA или нет?

Question

Я нахожу для функции типа type(), чтобы определить, какая переменная является CudaTensor или Normal. Метод

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