tf.pad函数

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

pad函数用于对张量进行填充，参数paddings表示填充的张量形式，参数constant_values表示填充值。mode表示填充的模式，有"CONSTANT", “REFLECT”, or "SYMMETRIC"三个可选值。

1、二维张量的例子

>>> a = tf.ones((2, 3))
>>> a

# [[2, 1], [1, 1]]表示在第一个维度上面填充2行，下面填充1行，在第二个维度左边填充1行，右边填充1行
>>> b = tf.pad(a, [[2, 1], [1, 1]])
>>> b

>>> c = tf.pad(a, [[0, 1], [1, 1]], constant_values=3)
>>> c

2、三维张量的例子

>>> a = tf.ones((3, 2, 3))
>>> a

# [[1, 2], [1, 1], [1, 1]]表示上面填充1行，下面填充2行，左边填充1行，右边填充1行
# 张量[1, 1, 1]里面左边填充1行，右边填充1行
>>> b = tf.pad(a, [[1, 2], [1, 1], [1, 1]])
>>> b