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

Commit 6add73c

Browse files
committed
Make Layer.inferring(from:) differentiable
1 parent 22c923a commit 6add73c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Sources/DeepLearning/Layer.swift

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ public extension Layer {
7878
///
7979
/// - Parameter input: The input to the layer.
8080
/// - Returns: The inference output.
81+
@differentiable(wrt: (self, input))
8182
func inferring(from input: Input) -> Output {
8283
let context = Context(learningPhase: .inference)
8384
return applied(to: input, in: context)

0 commit comments

Comments
 (0)