@tyoc213 see here: TypeError: expected CPU (got CUDA) when sublcassing / inheriting torch.Tensor class - #5 by turian