Function _param

test/backend/test_optim.py:13–14

(tensor, val)

m_init = np.random.randn(1,4).astype(np.float32)

def _param(tensor, val):
  return tensor(val, requires_grad=True) if tensor is torch.tensor else tensor(val)

class TeenyNet:
  def __init__(self, tensor):