DepthwiseConv2D

@frozen
public struct DepthwiseConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

שכבת פיתול דו-ממדית לעומק.

שכבה זו יוצרת מסנני קונבולציה ניתנים להפרדה שמתפתלים עם קלט השכבה כדי לייצר טנזור של יציאות.

  • ליבת הפיתול ה-4-D.

    הַצהָרָה

    public var filter: Tensor<Scalar>
  • וקטור ההטיה.

    הַצהָרָה

    public var bias: Tensor<Scalar>
  • פונקציית ההפעלה מבחינה אלמנטרית.

    הַצהָרָה

    @noDerivative
    public let activation: Activation
  • הצעדים של חלון הזזה לממדים מרחביים.

    הַצהָרָה

    @noDerivative
    public let strides: (Int, Int)
  • אלגוריתם הריפוד לקונבולציה.

    הַצהָרָה

    @noDerivative
    public let padding: Padding
  • סוג פונקציית ההפעלה מבחינת האלמנט.

    הַצהָרָה

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>
  • יוצר שכבת DepthwiseConv2D עם המסנן, ההטיה, פונקציית ההפעלה, הצעדים והריפוד שצוין.

    הַצהָרָה

    public init(
      filter: Tensor<Scalar>,
      bias: Tensor<Scalar>? = nil,
      activation: @escaping Activation = identity,
      strides: (Int, Int) = (1, 1),
      padding: Padding = .valid
    )

    פרמטרים

    filter

    ליבת הפיתול ה-4-D.

    bias

    וקטור ההטיה.

    activation

    פונקציית ההפעלה מבחינה אלמנטרית.

    strides

    הצעדים של חלון הזזה לממדים מרחביים.

    padding

    אלגוריתם הריפוד לקונבולציה.

  • מחזירה את הפלט המתקבל מהחלת השכבה על הקלט הנתון.

    הַצהָרָה

    @differentiable
    public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    פרמטרים

    input

    הקלט לשכבת הצורה, [ספירת אצווה, גובה קלט, רוחב קלט, ספירת ערוצי קלט]

    ערך החזרה

    הפלט של צורה, [ספירת אצווה, גובה פלט, רוחב פלט, ספירת ערוצי קלט * מכפיל ערוצים]

  • יוצר שכבת DepthwiseConv2D עם צורת המסנן שצוינה, הצעדים, הריפוד ופונקציית ההפעלה מבחינת האלמנט.

    הַצהָרָה

    public init(
      filterShape: (Int, Int, Int, Int),
      strides: (Int, Int) = (1, 1),
      padding: Padding = .valid,
      activation: @escaping Activation = identity,
      useBias: Bool = true,
      filterInitializer: ParameterInitializer<Scalar> = glorotUniform(),
      biasInitializer: ParameterInitializer<Scalar> = zeros()
    )

    פרמטרים

    filterShape

    הצורה של גרעין הקונבולציה ה-4-D עם צורה, [רוחב מסנן, גובה מסנן, ספירת ערוצי קלט, מכפיל ערוצים].

    strides

    הצעדים של חלון הזזה לממדים מרחביים/מרחביים-זמניים.

    padding

    אלגוריתם הריפוד לקונבולציה.

    activation

    פונקציית ההפעלה מבחינה אלמנטרית.

    filterInitializer

    אתחול לשימוש עבור פרמטרי המסנן.

    biasInitializer

    אתחול לשימוש עבור הפרמטרים ההטיה.