[Feature] adding tensor classes annotation for loss functions

[Feature] adding tensor classes annotation for loss functions #9261