Skip to content

Commit 4651ab3

Browse files
authored
Merge pull request #57 from hashina67/daily-task
Daily task
2 parents 889e1ed + 0022d40 commit 4651ab3

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

Tasks/daily tasks/Hashina/day_4.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
from torch import nn
2+
3+
class Network(nn.Module):
4+
def __init__(self):
5+
super().__init__()
6+
7+
self.hidden = nn.Linear(1, 2)
8+
self.output = nn.Linear(2, 1)
9+
10+
self.sigmoid = nn.Sigmoid()
11+
12+
def forward(self, x):
13+
x = self.hidden(x)
14+
x = self.sigmoid(x)
15+
x = self.output(x)
16+
17+
return x
18+

0 commit comments

Comments
 (0)