lua - Torch，如何检查变量是否是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