    def test_fuse_bn(self):
        """
        Test Fuse BN function.
        """
        conv = nn.Conv2d(in_channels=3, out_channels=3, kernel_size=1, stride=1, padding=0, dilation=1, groups=1)
        bn = nn.BatchNorm2d(3)
        f = fuse_bn(conv, bn)
        # print(len(f), f[0].shape, f[1].shape)
        self.assertTrue(len(f) == 2)
        self.assertTrue(f[0].shape == (3, 3, 1, 1))
        self.assertTrue(f[1].shape == (3,))
Test Fuse BN function.
tests/unittests/models/utils/test_fuse_bn.py