tf.raw_ops.Fill
Создает тензор, заполненный скалярным значением.
tf.raw_ops.Fill(
dims, value, name=None
)
Эта операция создает тензор формы dims и заполняет его значением value.
Например:
# Output tensor has shape [2, 3].
fill([2, 3], 9) ==> [[9, 9, 9]
[9, 9, 9]]
tf.fill отличается от tf.constant несколькими способами:
-
tf.fillподдерживает только скалярные значения, тогда какtf.constantподдерживает значения Tensor. -
tf.fillсоздает Op в графе вычислений, который формирует фактическое значение Tensor во время выполнения. Это отличается отtf.constant, который встраивает весь Tensor в граф с помощью узлаConst. - Поскольку
tf.fillвыполняется во время выполнения графа, он поддерживает динамические формы, зависящие от других Tensors времени выполнения, в отличие отtf.constant.
| Аргументы | |
|---|---|
dims | Объект типа Tensor. Должен иметь один из следующих типов: int32, int64. Одномерный. Задает форму выходного тензора. |
value | Объект типа Tensor. Нулевой размерности (скаляр). Значение для заполнения возвращаемого тензора. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Объект типа Tensor. Имеет тот же тип, что и value. |
Совместимость с numpy
Эквивалентно np.full
© 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/raw_ops/Fill