Update fractional_max_pool_op_test.py
This commit is contained in:
parent
271e5cf2cc
commit
170797b93a
@ -503,7 +503,7 @@ class FractionalMaxPoolGradTest(test.TestCase):
|
|||||||
input_data = self._GenerateUniqueRandomInputTensor(input_shape)
|
input_data = self._GenerateUniqueRandomInputTensor(input_shape)
|
||||||
# Add some randomness to make input_data not so 'integer'
|
# Add some randomness to make input_data not so 'integer'
|
||||||
input_data += self._PRNG.random_sample(input_shape)
|
input_data += self._PRNG.random_sample(input_shape)
|
||||||
pooling_ratio = [1, math.sqrt(13), math.sqrt(7), 1]
|
pooling_ratio = (1, math.sqrt(13), math.sqrt(7), 1)
|
||||||
output_shape = [int(a / b) for a, b in zip(input_shape, pooling_ratio)]
|
output_shape = [int(a / b) for a, b in zip(input_shape, pooling_ratio)]
|
||||||
overlapping = True
|
overlapping = True
|
||||||
pseudo_random = False
|
pseudo_random = False
|
||||||
|
Loading…
x
Reference in New Issue
Block a user