tf.fill
Создаёт тензор, заполненный скалярным значением.
tf.fill(
dims, value, name=None, layout=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.Tensors, в отличие от tf.constant(value, shape=dims), который встраивает значение как узел Const.
| Аргументы | |
|---|---|
dims | Последовательность одномерных неотрицательных чисел. Представляет форму выходного tf.Tensor. Элементы должны быть типа: int32, int64. |
value | Значение для заполнения возвращаемого tf.Tensor. |
name | Необязательная строка. Имя выходного tf.Tensor. |
layout | Необязательно, tf.experimental.dtensor.Layout. Если указано, результат — DTensor с указанным расположением. |
| Возвращаемое значение | |
|---|---|
tf.Tensor с формой dims и тем же типом, что и у value. |
| Исключения | |
|---|---|
InvalidArgumentError | dims содержит отрицательные значения. |
NotFoundError | dims содержит нецелые значения. |
Совместимость с NumPy
Аналогично np.full. В numpy поддерживается больше параметров. Передача числового аргумента в качестве формы (np.full(5, value)) допустима в numpy для указания результата с формой 1-D, в то время как TensorFlow не поддерживает эту синтаксическую конструкцию.
© 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/fill