Instantly share code, notes, and snippets.

Embed
What would you like to do?
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)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment