Building a neural layer in Swift