tensorflow中tf.nn.conv2d_transpose的应用

最新推荐文章于 2024-08-12 04:15:00 发布

UESTC_C2_403

最新推荐文章于 2024-08-12 04:15:00 发布

阅读量4.6k

点赞数

分类专栏： tensorflow用法

本文链接：https://blog.csdn.net/UESTC_C2_403/article/details/73385433

版权

tensorflow用法专栏收录该内容

64 篇文章 9 订阅

订阅专栏

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 tf

x = 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))