def testComputationStrided(self, use_bias):
"""Run through for something with a known answer."""
conv1 = snt.CausalConv1D(
output_channels=1,
kernel_shape=3,
stride=2,
use_bias=use_bias,
name="conv1",
initializers=create_constant_initializers(1.0, 1.0, use_bias))
out = conv1(tf.constant(np.ones([1, 5, 1], dtype=np.float32)))
expected_out = np.reshape(np.array([1, 3, 3]), [1, 3, 1])
if use_bias:
expected_out += 1
init_op = tf.variables_initializer(
[conv1.w, conv1.b] if use_bias else [conv1.w])
with self.test_session() as sess:
sess.run(init_op)
actual_out = sess.run(out)
self.assertAllClose(actual_out, expected_out)
评论列表
文章目录