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

Update test case and set non-zero imag part of the complex number

Browse Source 
from review comment feedback

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-01-21 17:24:15 +00:00
parent f3c49c5e45
commit 72317f192d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/kernel_tests/extract_image_patches_op_test.py
View File

@ -128,8 +128,12 @@ class ExtractImagePatches(test.TestCase):
def testComplexDataTypes(self):
"""Test for complex data types"""
for dtype in [np.complex64, np.complex128]:
image = np.reshape(range(120), [2, 3, 4, 5]).astype(dtype)
patches = np.reshape(range(120), [2, 3, 4, 5]).astype(dtype)
image = (
np.reshape(range(120), [2, 3, 4, 5]).astype(dtype) +
np.reshape(range(120, 240), [2, 3, 4, 5]).astype(dtype) * 1j)
patches = (
np.reshape(range(120), [2, 3, 4, 5]).astype(dtype) +
np.reshape(range(120, 240), [2, 3, 4, 5]).astype(dtype) * 1j)
for padding in ["VALID", "SAME"]:
self._VerifyValues(
image,