description: Like tf.expand_dims but accepts a vector of axes to expand.
Like tf.expand_dims but accepts a vector of axes to expand.
tf.expand_dims
tfp.experimental.nn.util.expand_dims( axis, name=None )