Modul

  • Jenis masukan lapisan.

    Pernyataan

    associatedtype Input
  • Jenis keluaran lapisan.

    Pernyataan

    associatedtype Output : Differentiable
  • Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

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

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Outputnya.

  • maju(_:) Implementasi bawaan

    Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Implementasi Bawaan

    Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

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

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Outputnya.

Tersedia di mana `Input`: `TensorProtocol`, `Output`: `DifferentiableTensorProtocol`
  • panggilanSebagaiFungsi(_:) Implementasi bawaan

    Implementasi Bawaan

    Mengembalikan keluaran beranotasi yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

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

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Output yang dianotasi.

  • beranotasi(_:) Metode penyuluhan

    Memberi anotasi pada output .

    Catatan: Mengembalikan output jika menggunakan backend yang tidak mendukung anotasi.

    Pernyataan

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

    Parameter

    output

    Output ke lapisan.

    Nilai Pengembalian

    Output yang dianotasi.

  • ringkasan (masukan :) Metode penyuluhan

    Mengembalikan anotasi yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

    public func summary(input: Input) -> String

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Semua anotasi dikumpulkan dari grafik XLA.