tf.shape(x)与tensor.get

mac2026-01-02  8

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

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

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

               a.get_shape()中a的数据类型只能是tensor,且返回的是一个元组(tuple)                tensor.get_shape().as_list()  将返回的元组重新变回一个列表(list)

最新回复(0)