Creating Custom Layers and Loss Functions in PyTorch

Creating custom layers and loss functions in