TensorFlow：tf.shape_n函数

tf.shape_n 函数

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

参见指南：张量变换>形状的确定与改变

返回张量的形状.

tf.shape_n函数操作返回表示input[i]s形状的N个1维整数张量.

函数参数

• input：至少1个Tensor对象具有相同类型的列表.
• out_type：可选的tf.DType,来自：tf.int32和tf.int64；默认为tf.int32.
• name：操作的名称(可选).

函数返回

与类型为out_type的Tensor对象的input具有相同长度的列表.