Modul

public protocol Module: EuclideanDifferentiable, KeyPathIterable
where
  TangentVector: VectorProtocol & ElementaryFunctions & PointwiseMultiplicative & KeyPathIterable

  • 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

    Hasil.

  • 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

    Hasil.

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 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.

, 
public protocol Module: EuclideanDifferentiable, KeyPathIterable
where
  TangentVector: VectorProtocol & ElementaryFunctions & PointwiseMultiplicative & KeyPathIterable

  • 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

    Hasil.

  • 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

    Hasil.

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 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.