Equivalent to torch.unsqueeze() in tensorflow

[kaiyuyue]

In TF, the tf.expand_dims() is equivalent to the torch.unsqueeze().