João C. Rodrigues Jr.

Fixing issue #57 related to cuda

1

When using cuda, the code is trying to take the argmin of an array that's being constructed with torch.Tensor objects, which results in an error that aborts the entire execution....