Spec-Zone.ru › TensorFlow

tf.pad

Заполняет тензор.

tf.pad(
    tensor, paddings, mode='CONSTANT', constant_values=0, name=None
)

Используется в блокнотах

Используется в руководстве Используется в учебных пособиях
  • Расширенная автоматическая дифференциация
  • Эффективное для клиентов обучение на больших моделях в федеративном обучении с помощью `federated_select` и разреженной агрегации
  • Обзор TensorFlow Probability
  • Обнаружение нескольких точек изменения и байесовский выбор модели
  • Обучение нейронным структурам на основе графов в TFX

Данная операция заполняет tensor согласно указанным вами paddings. paddings — это целочисленный тензор с формой [n, 2], где n — ранг tensor. Для каждого измерения D input, paddings[D, 0] указывает количество значений, добавляемых перед содержимым tensor в данном измерении, а paddings[D, 1] — количество значений, добавляемых после содержимого tensor в данном измерении. Если mode равно "REFLECT", то и paddings[D, 0], и paddings[D, 1] должны быть не больше tensor.dim_size(D) - 1. Если mode равно "SYMMETRIC", то и paddings[D, 0], и paddings[D, 1] должны быть не больше tensor.dim_size(D).

Размер заполненного каждого измерения D выходного тензора:

paddings[D, 0] + tensor.dim_size(D) + paddings[D, 1]

Например:

t = tf.constant([[1, 2, 3], [4, 5, 6]])
paddings = tf.constant([[1, 1,], [2, 2]])
# 'constant_values' is 0.
# rank of 't' is 2.
tf.pad(t, paddings, "CONSTANT")  # [[0, 0, 0, 0, 0, 0, 0],
                                 #  [0, 0, 1, 2, 3, 0, 0],
                                 #  [0, 0, 4, 5, 6, 0, 0],
                                 #  [0, 0, 0, 0, 0, 0, 0]]

tf.pad(t, paddings, "REFLECT")  # [[6, 5, 4, 5, 6, 5, 4],
                                #  [3, 2, 1, 2, 3, 2, 1],
                                #  [6, 5, 4, 5, 6, 5, 4],
                                #  [3, 2, 1, 2, 3, 2, 1]]

tf.pad(t, paddings, "SYMMETRIC")  # [[2, 1, 1, 2, 3, 3, 2],
                                  #  [2, 1, 1, 2, 3, 3, 2],
                                  #  [5, 4, 4, 5, 6, 6, 5],
                                  #  [5, 4, 4, 5, 6, 6, 5]]
Аргументы
tensor Tensor.
paddings Tensor типа int32.
mode Одно из "CONSTANT", "REFLECT" или "SYMMETRIC" (регистр не важен)
constant_values В режиме "CONSTANT" скалярное значение заполнения. Должно быть того же типа, что и tensor.
name Имя операции (необязательно).
Возвращаемое значение
Tensor. Имеет тот же тип, что и tensor.
Исключения
ValueError При неверном режиме (не "CONSTANT", "REFLECT" или "SYMMETRIC").

© 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/pad

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API