torch / nngraph

Graph Computation for nn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Automatic tensor type conversion

NanyangYe opened this issue · comments

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