wrong return type

[sjanssen2]

Hi Jamie,
I try to check the traditional hardmax NW but code execution fails unless I remove the , _ on the left hand side. I think the torch.max function only returns a single Tensor, doesn't it?

deepblast/deepblast/ops.py

Line 7 in ab48e4f

M, _ = torch.max(X)

[mortonjt]

Yes, you are right -- this looks like it was inherited from an older version of pytorch where max returned both the value and indices. This will need to updated.