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 | Последовательность одномерных неотрицательных чисел. Представляет форму возвращаемого 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 для указания одномерного результата, в то время как 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.4/api_docs/python/tf/fill