Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.nn.util.expand_dims

View source on GitHub

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

tfp.experimental.nn.util.expand_dims(
    axis, name=None
)