维度减少
tf.squeeze(input,axis=None,name=None):
input:输入张量;
axis:axis=1,表示要删掉的为1的维度。
代码:
生成一个大小为1001003的张量来表示一张尺寸为100*100的三通道彩色图片
squeeze_sample_1 = tf.random.normal([1,100,100,3])
print("原始数据尺寸:",squeeze_sample_1.shape)
squeezed_sample_1 = tf.squeeze(expand_sample_1)
print("维度压缩后的数据尺寸:",squeezed_sample_1.shape)
输出:
原始数据尺寸: (1, 100, 100, 3)
维度压缩后的数据尺寸: (100, 100, 3)