Lớp

Sử dụng bộ sưu tập để sắp xếp ngăn nắp các trang Lưu và phân loại nội dung dựa trên lựa chọn ưu tiên của bạn.
public protocol Layer : Module where Self.Input : Differentiable

Một lớp mạng nơ-ron.

Loại mà phù hợp với Layer đại diện cho chức năng mà đầu vào bản đồ để đầu ra. Chúng có thể có trạng thái bên trong được biểu thị bằng các thông số, chẳng hạn như bộ căng khối lượng.

Layer trường xác định một khả vi callAsFunction(_:) phương pháp đầu vào bản đồ để đầu ra.

  • Trả về kết quả thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    @differentiable
    func callAsFunction(_ input: Input) -> Output

    Thông số

    input

    Đầu vào cho lớp.

    Giá trị trả lại

    Đầu ra.

  • phía trước(_:)

    Triển khai mặc định

    Triển khai mặc định

    Tuyên ngôn

    @differentiable
    func forward(_ input: Input) -> Output
  • suy luận (từ :)

    Phương pháp mở rộng

    Trả về đầu ra suy luận thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    public func inferring(from input: Input) -> Output

    Thông số

    input

    Đầu vào cho lớp.

    Giá trị trả lại

    Kết quả suy luận.

  • Backpropagator

    Phương pháp mở rộng

    Tuyên ngôn

    public typealias Backpropagator = (_ direction: Output.TangentVector)
      -> (layerGradient: TangentVector, inputGradient: Input.TangentVector)
  • applyForBackpropagation (thành :)

    Phương pháp mở rộng

    Trả về đầu ra suy luận và hàm truyền ngược thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    public func appliedForBackpropagation(to input: Input)
      -> (output: Output, backpropagator: Backpropagator)

    Thông số

    input

    Đầu vào cho lớp.

    Giá trị trả lại

    Một bộ giá trị chứa đầu ra và chức năng nhân giống ngược. Hàm backpropagation (hay còn gọi là backpropagator) nhận một vectơ hướng và trả về các gradient ở lớp và ở đầu vào tương ứng.

Có sẵn ở nơi `Đầu vào`:` Khác biệtTensorProtocol`, `Đầu ra`:` Khác biệt

  • callAsFunction (_ :)

    Triển khai mặc định

    Triển khai mặc định

    Tuyên ngôn

    @differentiable(wrt: self)
    @differentiable
    public func callAsFunction(_ input: Input) -> Output