Yes, exactly, parametrized loss functions are rare and that is why you cannot find loss.cuda anywhere.



 samra-irshad:

loss functions which get some input tensors, apply loss function on those tensors and then give outputs.


Here, no need to use loss.cuda as its inputs all are cuda tensors.