Hello, bro, thank you for your code contribution. But I have a small suggestion, I think the ".view" in line 18 of the code in loss_objectives.py should be replaced with "t()". "view(-1,m)" in the source code will cause the mean to not be subtracted correctly.