Stop calling expand_1d on newer TF versions#20786

Rocketknight1 merged 1 commit intomainhuggingface/transformers:mainfrom

tf_train_step_for_211huggingface/transformers:tf_train_step_for_211Copy head branch name to clipboard