Sigmoid function or logistic function

sig.py

import torch

#----------- Implementing the math -----------#

def sigmoid(z):
  return 1 / (1+torch.exp(-z))

z = torch.tensor([[2.], [-3.]]) # Two neurons with different values

sig = sigmoid(z) 

#----------- Using Pytorch -----------#

torch_sig = torch.sigmoid(z)

#----------- Comparing outputs -----------#

print(f"Pytorch Sigmoid: \n{torch_sig} \nOur Sigmoid: \n{sig}")
'''
Out:
Pytorch Sigmoid: 
tensor([[0.8808],[0.0474]]) 
Our Sigmoid: 
tensor([[0.8808],[0.0474]])
'''