public protocol Module: EuclideanDifferentiable, KeyPathIterable
where
TangentVector: VectorProtocol & ElementaryFunctions & PointwiseMultiplicative & KeyPathIterable
-
The input type of the layer.
Declaration
associatedtype Input
-
The output type of the layer.
Declaration
associatedtype Output : Differentiable
-
forward(_:)
Returns the output obtained from applying the layer to the given input.
Default Implementation
Returns the output obtained from applying the layer to the given input.
Parameters
input
The input to the layer.
Return Value
The output.
-
callAsFunction(_:)
-
annotated(_:)
-
summary(input:)
Returns the annotations obtained from applying the layer to the given input.
Declaration
public func summary(input: Input) -> String
Parameters
input
The input to the layer.
Return Value
All collected annotations from the XLA graph.