experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test case for additional arg with extract_glimpse

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2018-03-13 17:58:03 +00:00
parent 2c409af18d
commit a8105c7d2a
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/python/kernel_tests/attention_ops_test.py
View File

@ -21,6 +21,7 @@ from __future__ import print_function
import numpy as np import numpy as np
from tensorflow.python.framework import constant_op from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes
from tensorflow.python.ops import array_ops from tensorflow.python.ops import array_ops
from tensorflow.python.ops import image_ops from tensorflow.python.ops import image_ops
from tensorflow.python.platform import test from tensorflow.python.platform import test
@ -195,5 +196,15 @@ class ExtractGlimpseTest(test.TestCase):
expected_rows=[None, None, None, 1, 2, 3, 4], expected_rows=[None, None, None, 1, 2, 3, 4],
expected_cols=[56, 57, 58, 59, 60]) expected_cols=[56, 57, 58, 59, 60])
def testGlimpseNoOverlapZero(self):
img = constant_op.constant(np.arange(25).reshape((1, 5, 5, 1)),
dtype=dtypes.float32)
with self.test_session():
result = image_ops.extract_glimpse(img, [3, 3], [[-2, 2]],
centered=False, normalized=False,
uniform_noise=False, noise="zero")
self.assertAllEqual(np.asarray([[0, 0, 0], [0, 0, 0], [0, 0, 0]]),
result.eval()[0, :, :, 0])
if __name__ == '__main__': if __name__ == '__main__':
test.main() test.main()