tf.fill
| Просмотреть исходный код на GitHub |
Создаёт тензор, заполненный скалярным значением.
tf.fill(
dims, value, name=None
)
См. также tf.ones, tf.zeros, tf.one_hot, tf.eye.
Данная операция создаёт тензор с формой dims и заполняет его значением value.
Например:
tf.fill([2, 3], 9)
<tf.Tensor: shape=(2, 3), dtype=int32, numpy=
array([[9, 9, 9],
[9, 9, 9]], dtype=int32)>
tf.fill вычисляется во время выполнения графа и поддерживает динамические формы, основанные на других значениях, получаемых во время выполнения, в отличие от tf.constant(value, shape=dims), который встраивает значение в виде узла Const.
| Аргументы | |
|---|---|
dims | Последовательность 1-D неотрицательных чисел. Представляет форму выходного tf.Tensor. Элементы должны быть типа: int32, int64. |
value | Значение для заполнения возвращаемого tf.Tensor. |
name | Необязательная строка. Имя выходного tf.Tensor. |
| Возвращаемое значение | |
|---|---|
tf.Tensor с формой dims и тем же типом данных, что и value. |
| Исключения | |
|---|---|
InvalidArgumentError | dims содержит отрицательные элементы. |
NotFoundError | dims содержит нецелые элементы. |
Совместимость с Numpy
Аналогично np.full. В numpy, поддерживается больше параметров. Передача числового аргумента в качестве формы (np.full(5, value)) допустима в numpy для задания результата формы 1-D, тогда как TensorFlow не поддерживает этот синтаксис.
© 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/fill