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

Commit 34573a2

Browse files
author
Marc Rasi
committed
turn off reshape layer for v0.2 (it crashes the stdlib compile)
1 parent 6f5b962 commit 34573a2

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

Sources/DeepLearning/Layer.swift

-18
Original file line numberDiff line numberDiff line change
@@ -620,21 +620,3 @@ public struct Flatten<Scalar: TensorFlowFloatingPoint>: Layer {
620620
return input.reshaped(to: [batchSize, remaining])
621621
}
622622
}
623-
624-
@_fixed_layout
625-
public struct Reshape<Scalar: TensorFlowFloatingPoint>: Layer {
626-
@noDerivative public let shape: Tensor<Int32>
627-
628-
public init(shape: Tensor<Int32>) {
629-
self.shape = shape
630-
}
631-
632-
public init(_ shape: TensorShape) {
633-
self.init(shape: Tensor(shape.dimensions))
634-
}
635-
636-
@differentiable
637-
public func applied(to input: Tensor<Scalar>, in _: Context) -> Tensor<Scalar> {
638-
return input.reshaped(toShape: shape)
639-
}
640-
}

0 commit comments

Comments
 (0)