โมดูล

  • ประเภทอินพุตของเลเยอร์

    คำประกาศ

    associatedtype Input
  • ประเภทเอาต์พุตของเลเยอร์

    คำประกาศ

    associatedtype Output : Differentiable
  • ส่งกลับผลลัพธ์ที่ได้จากการใช้เลเยอร์กับอินพุตที่กำหนด

    คำประกาศ

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

    พารามิเตอร์

    input

    อินพุตไปยังเลเยอร์

    ค่าส่งคืน

    เอาท์พุท

  • ซึ่งไปข้างหน้า(_:) การใช้งานเริ่มต้น

    ส่งกลับผลลัพธ์ที่ได้จากการใช้เลเยอร์กับอินพุตที่กำหนด

    การใช้งานเริ่มต้น

    ส่งกลับผลลัพธ์ที่ได้จากการใช้เลเยอร์กับอินพุตที่กำหนด

    คำประกาศ

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

    พารามิเตอร์

    input

    อินพุตไปยังเลเยอร์

    ค่าส่งคืน

    เอาท์พุท

มีให้โดยที่ `Input`: `TensorProtocol`, `Output`: `DifferentiableTensorProtocol`
  • โทรAsFunction(_:) การใช้งานเริ่มต้น

    การใช้งานเริ่มต้น

    ส่งคืนเอาต์พุตที่มีคำอธิบายประกอบที่ได้รับจากการใช้เลเยอร์กับอินพุตที่กำหนด

    คำประกาศ

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

    พารามิเตอร์

    input

    อินพุตไปยังเลเยอร์

    ค่าส่งคืน

    เอาต์พุตที่มีคำอธิบายประกอบ

  • มีคำอธิบายประกอบ(_:) วิธีการขยาย

    คำอธิบายประกอบ output

    หมายเหตุ: ส่งคืน output หากใช้แบ็กเอนด์ที่ไม่รองรับคำอธิบายประกอบ

    คำประกาศ

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

    พารามิเตอร์

    output

    ส่งออกไปยังเลเยอร์

    ค่าส่งคืน

    เอาต์พุตที่มีคำอธิบายประกอบ

  • สรุป (อินพุต :) วิธีการขยาย

    ส่งคืนคำอธิบายประกอบที่ได้รับจากการใช้เลเยอร์กับอินพุตที่กำหนด

    คำประกาศ

    public func summary(input: Input) -> String

    พารามิเตอร์

    input

    อินพุตไปยังเลเยอร์

    ค่าส่งคืน

    คำอธิบายประกอบที่รวบรวมทั้งหมดจากกราฟ XLA