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))