def test_he_normal(self):
tensor_shape = (5, 6, 4, 2)
with self.cached_session():
fan_in, _ = _compute_fans(tensor_shape)
std = np.sqrt(2. / fan_in)
self._runner(initializers.HeNormalV2(seed=123),
tensor_shape,
target_mean=0.,
target_std=std)
def test_he_normal(self):
tensor_shape = (5, 6, 4, 2)
with self.cached_session():
self._runner(initializers.HeNormalV2(seed=123), tensor_shape)