| --- |
| license: apache-2.0 |
| language: |
| - en |
| --- |
| # logic-gate |
|
|
| yes 102 parameters, not million, not thousand, one hundred and two. |
| its the most useless and weirdest neural network, but can be ok for testing out stuff. |
|
|
| ## Details |
|
|
| Tiny `12 → 4 → 4 → 4 → 2` network that simulates 10 logic gates from one set of weights. |
|
|
| Input is a one-hot gate selector plus `a` and `b`. Output is 2 neurons. |
|
|
| ## Use |
|
|
| ```bash |
| pip install torch safetensors |
| python example.py |
| ``` |
|
|
| ```python |
| from example import forward |
| |
| print(forward("XOR", 1, 0).tolist()) # [p(out1), p(out2)] |
| print(forward("CNOT", 1, 0).tolist()) # [p(a), p(a XOR b)] |
| ``` |
|
|
| Gates: `AND OR XOR NAND NOR XNOR NOT BUF CNOT SWAP` |
|
|
| - One-output gates: `[result, 1 - result]` |
| - `CNOT(a, b) = (a, a XOR b)` |
| - `SWAP(a, b) = (b, a)` |
