tf.shape(
input,
name=None,
out_type=tf.int32
)
返回张量的形状。这个操作返回一个表示输入形状的一维整数张量。
例如:
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
: 操作的指定输出类型 (int32
or int64
)(可选的).返回值:
原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/shape?hl=en