tensorflow版本2.x
tf.pad()函数用来对tensor某个维度进行填充或者复制,有3种用法:
- CONSTANT(填充常数),
- REFLECT(镜像复制),
- 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=(