tf.keras.backend.repeat_elements
| Просмотреть исходный код на GitHub |
Повторяет элементы тензора вдоль оси, как np.repeat.
tf.keras.backend.repeat_elements(
x, rep, axis
)
Если x имеет форму (s1, s2, s3) и axis равно 1, результат будет иметь форму (s1, s2 * rep, s3).
| Аргументы | |
|---|---|
x | Тензор или переменная. |
rep | Целое число Python, число повторений. |
axis | Ось, вдоль которой повторять. |
| Возвращаемое значение | |
|---|---|
| Тензор. |
Пример:
b = tf.constant([1, 2, 3])
tf.keras.backend.repeat_elements(b, rep=2, axis=0)
<tf.Tensor: id=70, shape=(6,), dtype=int32,
numpy=array([1, 1, 2, 2, 3, 3], 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/r1.15/api_docs/python/tf/keras/backend/repeat_elements