(self, input_param, input_shape)
| 45 | class TestFeedForward(unittest.TestCase): |
| 46 | @parameterized.expand(TEST_CASES_FEEDFORWARD) |
| 47 | def test_shape(self, input_param, input_shape): |
| 48 | net = FeedForward(**input_param) |
| 49 | with eval_mode(net): |
| 50 | result = net(torch.randn(input_shape)) |
| 51 | self.assertEqual(result.shape, input_shape) |
| 52 | |
| 53 | def test_gating_mechanism(self): |
| 54 | net = FeedForward(spatial_dims=2, dim=32, ffn_expansion_factor=2.0, bias=True) |
nothing calls this directly
no test coverage detected