tf.compat.v1.pad

Добавляет прокладку к тензору.

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

Эта операция добавляет прокладку к 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]]