tensorflow中tf.nn.conv2d_transpose的应用

来源:互联网 发布:大淘客cms加淘口令 编辑:程序博客网 时间:2024/05/20 05:29

tf.nn.conv2d_transpose(x, w, output_shape, strides, padding='SAME'):推荐使用x的数据格式为默认格式[batch, height, width, in_channels]。w是滤波器,输入分格式为[height, width, output_channels, in_channels]。output_shape就是输出数据的格式[batch, height, width, in_channels],strides就是滑动的步长。[1,h_l, w_l,1]。输出数据格式的height等于h_l乘以输入的height。另外一个参数也是一样的。

例如:

import tensorflow as tfx = tf.ones([10,4,4,5])w = tf.ones([2,2,2,5])y = tf.nn.conv2d_transpose(x, w, output_shape=[10,12,12,2], strides=[1,3,3,1], padding='SAME')with tf.Session() as sess:    sess.run(tf.global_variables_initializer())    print(sess.run(y))


阅读全文
0 0
原创粉丝点击