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 | Список как минимум из одного объекта 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
