get.shape

a.get_shape() #得到数组a的维度 例如啊[[1,2],[3,4]]  得到[2,2]

a[-1]表示数组a最后一维

tf.shape(a)和a.get_shape()比较

   相同点:都可以得到tensor a的尺寸

   不同点:tf.shape()中a 数据的类型可以是tensor, list, array

               a.get_shape()中a的数据类型只能是tensor,且返回的是一个元组(tuple)

猜你喜欢

转载自blog.csdn.net/qq_40614981/article/details/81224417
今日推荐