Mô-đun

  • Kiểu đầu vào của lớp.

    Tuyên ngôn

    associatedtype Input
  • Loại đầu ra của lớp.

    Tuyên ngôn

    associatedtype Output : Differentiable
  • 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(wrt: self)
    func callAsFunction(_ input: Input) -> Output

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Đầu ra.

  • phía trước(_:) Triển khai mặc định

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

    Triển khai mặc định

    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(wrt: self)
    func forward(_ input: Input) -> Output

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Đầu ra.

Có sẵn ở nơi `Đầu vào`: `TensorProtocol`, `Đầu ra`: `DifferentiableTensorProtocol`
  • callAsFunction(_:) Triển khai mặc định

    Triển khai mặc định

    Trả về đầu ra có chú thích thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

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

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Đầu ra có chú thích.

  • chú thích(_:) Phương pháp mở rộng

    Chú thích output .

    Lưu ý: Trả về output nếu sử dụng phần phụ trợ không hỗ trợ chú thích.

    Tuyên ngôn

    @differentiable
    public func annotated(_ output: Output) -> Output

    Thông số

    output

    Đầu ra cho lớp.

    Giá trị trả về

    Đầu ra có chú thích.

  • tóm tắt (đầu vào :) Phương pháp mở rộng

    Trả về các chú thích thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    public func summary(input: Input) -> String

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Tất cả các chú thích được thu thập từ biểu đồ XLA.