tf.compat.v1.expand_dims

Returns a tensor with a le