fix(distill): flatten MFTLoss labels along with the logits - #2645

fix(distill): flatten MFTLoss labels along with the logits#2645

SID-6921 wants to merge 2 commits into

NVIDIA:mainNVIDIA/Model-Optimizer:mainfrom

SID-6921:fix/mft-loss-flatten-labelsSID-6921/Model-Optimizer:fix/mft-loss-flatten-labelsCopy head branch name to clipboard