layerInit.swift

class Layer {
  
    let size: Int
    let activation: Activation

    init(size: Int, activation: Activation) {
        self.size = size
        self.activation = activation
    }

    static func input(size: Int) -> Layer {
        return Layer(size: size, activation: .none)
    }
    
    static func fullyConnected(size: Int, activation: Activation) -> Layer {
        return Layer(size: size, activation: activation)
    }
}