TensorFlow:tf.shape_n函数

2018-01-23 10:00 更新

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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号