TensorFlow張量變換函數(shù)：tf.shape

tf.shape 函數(shù)

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

定義在：tensorflow/python/ops/array_ops.py.

參見指南：張量變換>形狀的確定與改變

返回某個(gè)張量的形狀.

此操作返回表示input形狀的一維整數(shù)張量.

例如：

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.shape(t)  # [2, 2, 3]

函數(shù)參數(shù)

• input：表示一個(gè)Tensor或SparseTensor.
• name：操作的名稱(可選).
• out_type:(可選)操作的指定輸出類型((int32或int64))；默認(rèn)為tf.int32.

函數(shù)返回值

tf.shape 函數(shù)會(huì)返回out_type類型的Tensor.