维度增加
tf.expand_dims(input,axis,name=None):
input:输入张量;
axis:在第axis维度后增加一个维度。在输入D尺寸的情况下,轴必须在[-(D + 1),D](含)范围内。负数代表倒序。
代码:
生成一个大小为1001003的张量来表示一张尺寸为100*100的三通道彩色图片
expand_sample_1 = tf.random.normal([100,100,3], seed=1)
print("原始数据尺寸:",expand_sample_1.shape)
print("在第一个维度前增加一个维度(axis=0):",tf.expand_dims(expand_sample_1, axis=0).shape)
print("在第二个维度前增加一个维度(axis=1):",tf.expand_dims(expand_sample_1, axis=1).shape)
print("在最后一个维度后增加一个维度(axis=-1):",tf.expand_dims(expand_sample_1, axis=-1).shape)
输出:
原始数据尺寸: (100, 100, 3)
在第一个维度前增加一个维度(axis=0): (1, 100, 100, 3)
在第二个维度前增加一个维度(axis=1): (100, 1, 100, 3)
在最后一个维度后增加一个维度(axis=-1): (100, 100, 3, 1)