Error about devices

[Harr7y]

grad_input = k * t * (1 - torch.pow(torch.tanh(input * t), 2)) * grad_output

Expected all tensors to be on the same device, but found at least two devices , cuda:1 and cuda:0!