tensorflow中的tf.pad()函数的用法

tensorflow版本2.x
tf.pad()函数用来对tensor某个维度进行填充或者复制，有3种用法：

1. CONSTANT（填充常数），
2. REFLECT（镜像复制），
3. SYMMETRIC（对称复制）；

用法

CONSTANT

t = tf.constant([[1, 2, 3], [4, 5, 6]])
paddings = tf.constant([[1, 2], [5, 2]])
print(t)
t2 = tf.pad(t, paddings, "CONSTANT")
print(t2)

运行结果：

tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
<tf.Tensor: shape=(5, 10