Method __init__

exir/tests/models.py:25–26

(self)

class TensorItem(nn.Module):
    def __init__(self) -> None:
        super().__init__()

    def forward(self, arg1: torch.Tensor, arg2: torch.Tensor) -> torch.Tensor:
        h = arg1.item()