Method __init__

test/backend/test_optim.py:24–27

(self, tensor)

class TinyNet:
  def __init__(self, tensor):
    self.x = _param(tensor, x_init.copy())
    self.W = _param(tensor, W_init.copy())
    self.m = tensor(m_init.copy())

  def forward(self):
    out = self.x.matmul(self.W).relu()