Expand treatment indicator tensor to match the logits and predictions rank.
PiperOrigin-RevId: 608710316
N
Nicolas Aagnes committed
35dc5d0c460f7c81c37ce850c75491c12c37c810
Parent: efced1f
Committed by A. Unique TensorFlower <gardener@tensorflow.org>
on 2/20/2024, 9:07:55 PM