トーチの文法説明[CPUTYPE]

Question

私は最初のファイルdataloader.lua of fb.resnest.torchにtorch[cpuType]ようLUAでの使用を参照してください。

batch = torch[cpuType](sz, nCrops, table.unpack(imageSize)) 

私はそれのいずれかの文法の説明を見つけることができませんでした。それを理解する方法？

PS：cpuTypeがファイル内に定義されています。すなわち、self.cpuTypeと思います。

更新：私のテストから、torch['FloatTensor']はtorch.FloatTensorに相当します。

Answer 1

torch[cpuType]はtorch.cpuTypeと同じだと思います。

コード（https://github.com/facebook/fb.resnet.torch/blob/master/dataloader.lua#L51-L57）はcpuType、すなわちDoubleTensor、FloatTensorまたはHalfTensor、いくつかの異なる値を取ることができることを言っているようです。したがって、この表記法はtorch.DoubleTensorまたはtorch.FloatTensorまたはtorch.HalfTensorを作成します。

if cpuType == 'torch.DoubleTensor' then 
    batch = torch.DoubleTensor(sz, nCrops, table.unpack(imageSize)) 
elseif cpuType == 'torch.FloatTensor' then 
    batch = torch.FloatTensor(sz, nCrops, table.unpack(imageSize)) 
elseif cpuType == 'torch.HalfTensor' then 
    batch = torch.HalfTensor(sz, nCrops, table.unpack(imageSize)) 

Answer 2

ルアトーチ（私もルアトーチも試しました）と非常によく似ている私の知るところでは、このテンソルをどこに保存するかを指定しています。トーチは2つの異なる処理ユニットに格納された操作を実行できないことに注意してください。 cpu（net≈c.cu（））とgpu [net.cuda（）]の間でデータを移動する方法があります。