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 для указания результата одномерной формы, но 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