Which loss function is suitable?

Question

I am using PyTorch and am still quite new to the library.

I have a relation between my input and output given by y = ax + b, where a and b