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

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Tue Oct 2 13:23:27 2018
@author: myhaspl
@email:myhaspl@myhaspl.com
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
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
[[[[ 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.]]]]

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！