TensorFlow:tf.shape_n函数

由 Carrie 创建, 最后一次修改 2018-01-23

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具有相同长度的列表.

以上内容是否对您有帮助:

二维码
建议反馈
二维码