x=tf.constant([[1,2,3],[4,5,6]])
x.get_shape().as_list()[0] #输出:行数
x.get_shape().as_list()[1] #输出:列数
as_list()极简用法
最新推荐文章于 2021-10-19 10:17:14 发布
x=tf.constant([[1,2,3],[4,5,6]])
x.get_shape().as_list()[0] #输出:行数
x.get_shape().as_list()[1] #输出:列数