Spec-Zone.ru › TensorFlow 2.3

tf.repeat

Просмотреть исходный код на GitHub

Повторяет элементы input.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.repeat

tf.repeat(
    input, repeats, axis=None, name=None
)

См. также tf.concat, tf.stack, tf.tile.

Аргументы
input N-мерный тензор.
repeats Одномерный int тензор. Количество повторений для каждого элемента. repeats транслируется для соответствия форме заданной оси. len(repeats) должно быть равно input.shape[axis], если ось не равна None.
axis Целое число. Ось, вдоль которой нужно повторять значения. По умолчанию (axis=None), используется уплощенный входной массив, и возвращается плоский выходной массив.
name Имя операции.
Возвращаемое значение
Тензор, имеющий такую же форму, как input, за исключением заданной оси. Если ось равна None, то выходной массив уплощается для соответствия уплощенному входному массиву.

Пример использования:

repeat(['a', 'b', 'c'], repeats=[3, 0, 2], axis=0)
<tf.Tensor: shape=(5,), dtype=string,
numpy=array([b'a', b'a', b'a', b'c', b'c'], dtype=object)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=0)
<tf.Tensor: shape=(5, 2), dtype=int32, numpy=
array([[1, 2],
       [1, 2],
       [3, 4],
       [3, 4],
       [3, 4]], dtype=int32)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=1)
<tf.Tensor: shape=(2, 5), dtype=int32, numpy=
array([[1, 1, 2, 2, 2],
       [3, 3, 4, 4, 4]], dtype=int32)>
repeat(3, repeats=4)
<tf.Tensor: shape=(4,), dtype=int32, numpy=array([3, 3, 3, 3], dtype=int32)>
repeat([[1,2], [3,4]], repeats=2)
<tf.Tensor: shape=(8,), dtype=int32,
numpy=array([1, 1, 2, 2, 3, 3, 4, 4], dtype=int32)>

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/repeat

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API