Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit 3efbf45

Browse files
author
Marc Rasi
committed
add public initializer for Flatten
1 parent 34573a2 commit 3efbf45

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Sources/DeepLearning/Layer.swift

+2
Original file line numberDiff line numberDiff line change
@@ -613,6 +613,8 @@ public struct UpSampling2D<Scalar: TensorFlowFloatingPoint>: Layer {
613613

614614
@_fixed_layout
615615
public struct Flatten<Scalar: TensorFlowFloatingPoint>: Layer {
616+
public init() {}
617+
616618
@differentiable
617619
public func applied(to input: Tensor<Scalar>, in _: Context) -> Tensor<Scalar> {
618620
let batchSize = input.shape[0]

0 commit comments

Comments
 (0)