tf.nn.conv2d

Posted hapyygril

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tf.nn.conv2d相关的知识，希望对你有一定的参考价值。

import numpy as np
filter_primes = np.array([2., 3., 5., 7., 11., 13.], dtype=np.float32)
x = tf.constant(np.arange(1, 13+1, dtype=np.float32).reshape([1, 1, 13, 1]))
filters = tf.constant(filter_primes.reshape(1, 6, 1, 1))

valid_conv = tf.nn.conv2d(x, filters, strides=[1, 1, 5, 1], padding=‘VALID‘)
same_conv = tf.nn.conv2d(x, filters, strides=[1, 1, 5, 1], padding=‘SAME‘)

with tf.Session() as sess:
       print("VALID:\n", valid_conv.eval())
       print("SAME:\n", same_conv.eval())

以上是关于tf.nn.conv2d的主要内容，如果未能解决你的问题，请参考以下文章