Save the date! Google I/O returns May 18-20 Register now

tfp.experimental.nn.util.expand_dims

Like tf.expand_dims but accepts a vector of axes to expand.