tf.repeat
| Просмотреть исходный код на GitHub |
Повторяет элементы input.
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)>
© 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/versions/r2.9/api_docs/python/tf/repeat