public protocol Layer : Module where Self.Input : Differentiable
שכבת רשת עצבית.
סוגים שמתאים Layer
לייצג פונקציות כי תשומות המפה כדי תפוקות. יכול להיות שיש להם מצב פנימי המיוצג על ידי פרמטרים, כגון מתיחות משקל.
Layer
מקרים להגדיר גזיר callAsFunction(_:)
שיטת תשומות מיפוי יציאות.
מחזירה את הפלט המתקבל מיישום השכבה על הקלט הנתון.
הַצהָרָה
@differentiable func callAsFunction(_ input: Input) -> Output
פרמטרים
input
הקלט לשכבה.
ערך החזרה
הפלט.
יישום ברירת מחדל
הַצהָרָה
@differentiable func forward(_ input: Input) -> Output
מחזירה את פלט המסקנה המתקבל מיישום השכבה על הקלט הנתון.
הַצהָרָה
public func inferring(from input: Input) -> Output
פרמטרים
input
הקלט לשכבה.
ערך החזרה
פלט המסקנה.
הַצהָרָה
public typealias Backpropagator = (_ direction: Output.TangentVector) -> (layerGradient: TangentVector, inputGradient: Input.TangentVector)
מחזירה את פלט ההסקה ואת פונקציית ההפצה האחורית המתקבלת מיישום השכבה על הקלט הנתון.
הַצהָרָה
public func appliedForBackpropagation(to input: Input) -> (output: Output, backpropagator: Backpropagator)
פרמטרים
input
הקלט לשכבה.
ערך החזרה
צמד המכיל את הפלט ואת פונקציית ההפצה האחורית. פונקציית ההפצה האחורית (aka backpropagator) לוקחת וקטור כיוון ומחזירה את שיפועי השכבה ובכניסה, בהתאמה.
יישום ברירת מחדל
הַצהָרָה
@differentiable(wrt: self) @differentiable public func callAsFunction(_ input: Input) -> Output