tf.experimental.numpy.expand_dims