Maybe add that this is loss is not normalized over the samples. Maybe we should add a boolean option to make normalization possible?