TensorFlow中expand_dim()

expand_dim()：

作用：

• 使用tf.expand_dims(input, axis=None, name=None, dim=None)函数来给矩阵增加一个维度。

例子：

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

说明：

• axis用来指定增加的一个维度中的元素包含的维度