Method __init__

imperative/python/test/unit/utils/test_module_stats.py:110–112

(self)

def test_tm_get_weights():
    class Net(M.Module):
        def __init__(self):
            super().__init__()
            self.weight = mge.tensor(np.random.randn(3, 3))

        def forward(self, x):
            return x * self.weight