Jestli to dobře chápu, tak pokud se mají výpočty provádět na GPU, musí se to nastavit při konstrukci tenzoru že? Funguje to i u obrázků? protože toto je asi nejtypičtější použití GPU ne? Jinak ten jiný testovací obrázek pobavit, už vidím, jak někdo učí svoji NN na úplně jinej vstup :-)
jde to velmi jednoduše: https://kbullaughey.github.io/lstm-play/2015/09/21/torch-and-gpu.html
potom:
t=torch.CudaTensor
t:getDevice()
popř. transfer na GPU:
target = source:cuda()