代码语言：javascript复制
tf.shape(
input,
name=None,
out_type=tf.int32
)返回张量的形状。这个操作返回一个表示输入形状的一维整数张量。
例如:
代码语言：javascript复制t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.shape(t) # [2, 2, 3]参数:
input: 张量或稀疏张量。name: 操作的名字(可选的)。out_type: 操作的指定输出类型 (int32orint64)(可选的).
返回值:
- 一个类型为out_type的张量。
原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/shape?hl=en
