Working on NormalizeAndBroadcast in OnlineCFTrainHybridKernel