TensorFlow：tf.shape_n函數(shù)

tf.shape_n 函數(shù)

shape_n(
    input,
    out_type=tf.int32,
    name=None
)

參見指南：張量變換>形狀的確定與改變

返回張量的形狀.

tf.shape_n函數(shù)操作返回表示input[i]s形狀的N個1維整數(shù)張量.

函數(shù)參數(shù)

• input：至少1個Tensor對象具有相同類型的列表.
• out_type：可選的tf.DType,來自：tf.int32和tf.int64；默認為tf.int32.
• name：操作的名稱(可選).

函數(shù)返回

與類型為out_type的Tensor對象的input具有相同長度的列表.