@frozen
public struct GroupNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Lapisan yang menerapkan normalisasi grup pada sejumlah kecil masukan.
Referensi: Normalisasi Grup .
Nilai offset, juga dikenal sebagai beta.
Pernyataan
public var offset: Tensor<Scalar>
Nilai skala, juga dikenal sebagai gamma.
Pernyataan
public var scale: Tensor<Scalar>
Jumlah kelompok.
Pernyataan
@noDerivative public let groupCount: Int
Sumbu tempat fitur berada.
Pernyataan
@noDerivative public let axis: Int
Nilai varians epsilon.
Pernyataan
@noDerivative public let epsilon: Scalar
Membuat lapisan normalisasi grup.
Prasyarat
Sumbu tidak boleh berupa sumbu batch.Prasyarat
Offset harus memiliki peringkat 1.Prasyarat
Banyaknya elemen offset harus habis dibagi kelompok.Prasyarat
Offset dan skalanya harus mempunyai bentuk yang sama.Pernyataan
Parameter
offset
Nilai offset awal.
scale
Nilai skala awal.
groupCount
Jumlah kelompok.
axis
Sumbu tempat fitur berada.
epsilon
Nilai varians epsilon.
Membuat lapisan normalisasi grup.
Prasyarat
Sumbu tidak boleh berupa sumbu batch.Prasyarat
Jumlah fitur harus dapat dibagi berdasarkan kelompok.Pernyataan
public init( featureCount: Int, groupCount: Int, axis: Int = -1, epsilon: Scalar = 1e-3 )
Parameter
featureCount
Jumlah fitur.
groupCount
Jumlah kelompok.
axis
Sumbu tempat fitur berada. Nilai defaultnya adalah -1.
epsilon
Skalar kecil menambah varians. Nilai defaultnya adalah 0,001.
Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.
Prasyarat
Sumbu tidak boleh berupa sumbu batch.Prasyarat
Jumlah fitur masukan dan offset harus sama.Parameter
input
Masukan ke lapisan.
Nilai Pengembalian
Outputnya.