tf.shape_n
Возвращает форму списка тензоров.
tf.shape_n(
input,
out_type=tf.dtypes.int32,
name=None
)
Для данного списка тензоров, tf.shape_n значительно быстрее, чем применение tf.shape к каждому тензору по отдельности.
>>> a = tf.ones([1, 2]) >>> b = tf.ones([2, 3]) >>> c = tf.ones([3, 4]) >>> tf.shape_n([a, b, c]) [<tf.Tensor: shape=(2,), dtype=int32, numpy=array([1, 2], dtype=int32)>, <tf.Tensor: shape=(2,), dtype=int32, numpy=array([2, 3], dtype=int32)>, <tf.Tensor: shape=(2,), dtype=int32, numpy=array([3, 4], dtype=int32)>]
| Аргументы | |
|---|---|
input | Список из как минимум 1 Tensor объекта с одинаковым типом данных. |
out_type | Указанный тип результата операции (int32 или int64). По умолчанию tf.int32(необязательно). |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Список Tensor, определяющий форму каждого входного тензора с типом out_type. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/shape_n
