Issue with tensor: RuntimeError element 0 of tensors does not require grad and does not have a grad_fn

[Palash_Shah]

@Torch tensor reshape by removing a channel vision

output,_ = torch.max(output, 1, keepdim=True)

so, outputs = torch.argmax(outputs, 1, keepdim=True) ?