How to explicitly broadcast a tensor to match another's shape in tensorflow?

Question

I have three tensors, A, B and C in tensorflow, A and B are both of shape (m, n, r), C is a binary tensor of sha

Answer 1

In the newest tensorflow version(2.0), you can use tf.broadcast_to as below:

import tensorflow as tf

A = tf.random_normal([20, 100, 10])
B = tf.random_normal([20, 100, 10])
C = tf.random_normal([20, 100, 1])
C = tf.greater_equal(C, tf.zeros_like(C))
C = tf.broadcast_to(C, A.shape)

D = tf.where(C,A,B)