code: def grad(self, xs, ys):
          
          return self.sess.run(
              self.cross_entropy_grads, feed_dict={
                  self.x: xs,
                  self.y_: ys
              })
docs: Computes the gradients of the network.