tf.repeat
| Просмотреть исходный код на GitHub |
Повторяет элементы input.
tf.repeat(
input, repeats, axis=None, name=None
)
См. также tf.concat, tf.stack, tf.tile.
| Аргументы | |
|---|---|
input | Тензор размерности N. |
repeats | 1-мерный 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.4/api_docs/python/tf/repeat