feed-forward neural networks y_train shapetorch size([ ]y_test shapetorch size([ ] dtype torch float dtype torch float the toy datasetwith , samples each having featuresis divided into train and test let' create class that defines the neural network using pytorch' nn module listing - defines the creation of neural network for the purpose of this exercise listing - defining feed forward neural network #define neural network with hidden layers and output layer #hidden layers will have , and neurons #output layers will have neuron class neuralnetwork(nn module)def __init__(self)super(__init__(tch manual_seed( self fc nn linear( self relu nn relu(self fc nn linear( self relu nn relu(self out nn linear( self final nn sigmoid(def forward(selfx)op self fc (xop self relu (opop self fc (opop self relu (op