Views
No views yet
| sum | out |
|---|---|
| 0 | 0 |
| 1 | 1 |
| 2 | 0 |
| 3 | 0 |
| 4 | 0 |
| 5 | 0 |
| Inputs | 5 |
| Outputs | 1 |
| Neurons | 3 |
| Layers | 2 |
| Parameters | 15 |
| Magnitude | 16 |
1from safetensors.torch import load_file
2import torch
3
4w = load_file('model.safetensors')
5
6def exactly1of5(a, b, c, d, e):
7 inp = torch.tensor([float(a), float(b), float(c), float(d), float(e)])
8 l1 = (inp @ w['layer1.weight'].T + w['layer1.bias'] >= 0).float()
9 out = (l1 @ w['layer2.weight'].T + w['layer2.bias'] >= 0).float()
10 return int(out.item())
11
12print(exactly1of5(1, 0, 0, 0, 0)) # 1 (sum=1)
13print(exactly1of5(1, 1, 0, 0, 0)) # 0 (sum=2)