https://blog.csdn.net/yy_diego/article/details/81563160
https://www.tensorflow.org/api_docs/python/tf/pad
1.二维张量
pad( tensor, paddings, mode='CONSTANT',#constant默认就是0 name=None )
三行四列的1值,[1,2]表示对第一维,也就是行,第一行上面添加一行0,下面添加两行0,[3,4]对第二个维度,左边添加3列0,右边添加4列0.
# 创建一个二维变量,默认执行CONSTANT填充 vct = tf.Variable(tf.ones([3, 4]), name="vct") # 指定填充方式, pad1 = np.array([[1, 2], [3, 4]]) # tf.pad进行填充 vct_pad1 = tf.pad(vct, pad1, name='pad_1') # 创建会话 with tf.Session() as sess: sess.run(tf.global_variables_initializer()) print(sess.run(vct)) print(sess.run(vct_pad1))
2.三维张量
//这个就更好理解了。
tsr = tf.Variable(tf.ones([2, 3, 4]), name="tsr") pad2 = np.array([[1, 1], [0, 0], [0, 0]]) tsr_pad2 = tf.pad(tsr, pad2, name='pad_2') with tf.Session() as sess: sess.run(tf.global_variables_initializer()) print(sess.run(tsr)) print(sess.run(tsr_pad2))
[0,0]针对这个2维度,上下增加一个0,