torch / nngraph

Graph Computation for nn
Other
299 stars 96 forks source link

Automatic tensor type conversion #154

Open NanyangYe opened 6 years ago

NanyangYe commented 6 years ago

I found a problem when I use the cuda tensor as the default one. Even when I set: torch.set_default_tensor_type('torch.cuda.FloatTensor') input x (numpy.float) y (torch.cuda.tensor) output x*y is torch.float.tensor

Is this a feature or a bug. It might be more convenient if the programme behaves as the default setting. Many thanks