(self)
| 9 | |
| 10 | class LinearGen: |
| 11 | def __init__(self): |
| 12 | self.l1 = Tensor.scaled_uniform(128, 256) |
| 13 | self.l2 = Tensor.scaled_uniform(256, 512) |
| 14 | self.l3 = Tensor.scaled_uniform(512, 1024) |
| 15 | self.l4 = Tensor.scaled_uniform(1024, 784) |
| 16 | |
| 17 | def forward(self, x): |
| 18 | x = x.dot(self.l1).leaky_relu(0.2) |
nothing calls this directly
no test coverage detected