tf.shape()_Zhangppeng的专栏-CSDN博客_tf.shape
https://blog.csdn.net/apengpengpeng/article/details/8057965805.06.2018 · tf.shape tf.shape( input, name=None, out_type=tf.int32 ) 例如: 将矩阵的维度输出为一个维度矩阵 import tensorflow as tf import numpy as np A = np.array([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]], [[5, 5, 5], [6, 6, 6]]]) t = tf.shape(A) with tf.Session() as sess: