1. 程式人生 > >tensorflow隨筆-tf.nn.conv2d卷積運算(3)

tensorflow隨筆-tf.nn.conv2d卷積運算(3)

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Tue Oct  2 13:23:27 2018

@author: myhaspl
@email:[email protected]
tf.nn.conv2d
"""

import tensorflow as tf

g=tf.Graph()

with g.as_default():
    x=tf.constant([
            [[[1.],[2.]],[[3.],[4.]],[[5.],[6.]]],
            [[[10.],[20
.]],[[30.],[40.]],[[50.],[60.]]] ]) kernel=tf.constant([[[[2.,3.]]]]) y=tf.nn.conv2d(x,kernel,strides=[1,1,1,1],padding="SAME") with tf.Session(graph=g) as sess: print sess.run(x) print x.get_shape() print sess.run(y)
[[[[ 1.]
   [ 2.]]

  [[ 3.]
   [ 4.]]

  [[ 5.]
   [ 6.]]]


 [[[10.]
   [20.]]

  [[30.]
   [40.]]

  [[50.]
   [60.]]]]
(2, 3, 2, 1)
[[[[  2.   3.]
   [  4.   6.]]

  [[  6.   9.]
   [  8.  12.]]

  [[ 10.  15.]
   [ 12.  18.]]]


 [[[ 20.  30.]
   [ 40.  60.]]

  [[ 60.  90.]
   [ 80. 120.]]

  [[100. 150.]
   [120. 180.]]]]