add public initializer for Flatten

tensorflow · Feb 28, 2019 · 3efbf45 · 3efbf45
1 parent 34573a2
commit 3efbf45
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Sources/DeepLearning/Layer.swift b/Sources/DeepLearning/Layer.swift
@@ -613,6 +613,8 @@ public struct UpSampling2D<Scalar: TensorFlowFloatingPoint>: Layer {
 
 @_fixed_layout
 public struct Flatten<Scalar: TensorFlowFloatingPoint>: Layer {
+    public init() {}
+
     @differentiable
     public func applied(to input: Tensor<Scalar>, in _: Context) -> Tensor<Scalar> {
         let batchSize = input.shape[0]