From 81732b1f21d203c75f3d62109e767bae3601f0c0 Mon Sep 17 00:00:00 2001 From: Sourav Menon <45852906+souravmenon@users.noreply.github.com> Date: Wed, 17 Jun 2020 15:36:35 +0530 Subject: [PATCH] Task_3 --- Tasks/daily tasks/Sourav_Menon/task3.py | 30 +++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 Tasks/daily tasks/Sourav_Menon/task3.py diff --git a/Tasks/daily tasks/Sourav_Menon/task3.py b/Tasks/daily tasks/Sourav_Menon/task3.py new file mode 100644 index 0000000..c9e4796 --- /dev/null +++ b/Tasks/daily tasks/Sourav_Menon/task3.py @@ -0,0 +1,30 @@ +import torch +import torch as nn + +#creating a class as Net +class Net(nn.module): + + def __init__(self,input_layer,hidden_layer1,hidden_layer2,output_layer): + + super(Net,self).__init__() + self.t1=nn.Linear(input_size,hiden_layer1) + + self.f1=nn.sigmoid() + + self.t2=nn.Linear(hidden_layer1,hiddenl_layer2) + + self.f2=nn.sigmoid() + + self.t3=nn.linear(hidden_layer2,output_layer) + + + def forward(self,x): + x=self.t1(x) + x=self.f1(x) + x=self.t2(x) + x=self.f2(x) + x=self.t3(x) + + return x + +