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