python np.mean(axis) tf.squeeze(x)

python np.mean(axis) tf.squeeze(x)

精辟的一句话就是:axis等于哪一维,那一维被压缩为1(也就是不显示了。)


tensorflow的squeeze()函数则是默认压缩维度为1的那一维。也可通过squeeze_dim=[]来指定压缩的具体维度。