tf.expand_dims() 的用法

tf.expand_dims() 增加张量的维度

[ ]代表维度，由外向里数维度。

import tensorflow as tf

t = tf.constant([1,2]) # 创建一个张量
t

<tf.Tensor: shape=(2,), dtype=int32, numpy=array([1, 2])>

t1 = tf.expand_dims(t, 0) # 在第1个维度加1维（就是在第1维外加[]）
t1

<tf.Tensor: shape=(1, 2), dtype=int32, numpy=array([[1, 2]])>

t2 = tf.expand_dims(t, 1) # 在第2个维度加1维（就是在第2维外加[]）
t2

<tf.Tensor: shape=(2, 1), dtype=int32, numpy=array([[1],[2]])>

t3 = tf.expand_dims(t, -1) # 在最后1个维度加1维（就是在最后1维外加[]）
t3

<tf.Tensor: shape=(2, 1), dtype=int32, numpy=array([[1],[2]])>