Python tf.pad用法及代碼示例

填充張量。

用法

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

此操作根據您指定的 paddings 填充 tensor。 paddings 是形狀為 [n, 2] 的整數張量，其中 n 是 tensor 的秩。對於每個維度，input , paddings[D, 0]的D表示該維度中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]]