def test_layer_output_shape_after_forward_activation(self):
     layer = Dense(10, sigmoid)
     layer.build(20)
     layer.forward_activation(np.random.randn(20, 35))
     self.assertEqual(layer.A.shape, (10, 35))