tf.expand

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

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

import tensorflow as tf

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

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

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

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