tensorflow::ops::FusedBatchNormV2
#include <nn_ops.h>Normalisasi batch.
Ringkasan
Perlu dicatat bahwa ukuran Tensor 4D ditentukan oleh "NHWC" atau "NCHW". Ukuran Tensor 1D sesuai dengan dimensi C dari Tensor 4D.
Argumen:
- Cakupan: Objek Cakupan
- x: Tensor 4D untuk data masukan.
- skala: Tensor 1D untuk faktor penskalaan, untuk menskalakan x yang dinormalisasi.
- offset: Tensor 1D untuk offset, untuk menggeser ke x yang dinormalisasi.
- mean: Tensor 1D untuk rata-rata populasi. Digunakan hanya untuk inferensi; harus kosong untuk pelatihan.
- varians: Tensor 1D untuk varians populasi. Digunakan hanya untuk inferensi; harus kosong untuk pelatihan.
Atribut opsional (lihat Attrs ):
- epsilon: Sebuah angka desimal kecil yang ditambahkan ke varians x.
- data_format: Format data untuk x dan y. Bisa "NHWC" (default) atau "NCHW".
- is_training: Nilai boolean untuk menunjukkan apakah operasi tersebut untuk pelatihan (default) atau inferensi.
Mengembalikan:
-
Outputy: Tensor 4D untuk data keluaran. -
Outputbatch_mean: Tensor 1D untuk rata-rata batch yang dihitung, yang akan digunakan oleh TensorFlow untuk menghitung rata-rata berjalan. -
Outputbatch_variance: Tensor 1D untuk varians batch yang dihitung, yang akan digunakan oleh TensorFlow untuk menghitung varians berjalan. -
Outputreserve_space_1: Tensor 1D untuk rata-rata batch yang dihitung, untuk digunakan kembali dalam perhitungan gradien. -
Outputreserve_space_2: Tensor 1D untuk varians batch yang dihitung (varians terbalik dalam kasus cuDNN), yang akan digunakan kembali dalam perhitungan gradien.
Konstruktor dan Destruktor | |
|---|---|
FusedBatchNormV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance) | |
FusedBatchNormV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance, const FusedBatchNormV2::Attrs & attrs) |
Atribut publik | |
|---|---|
batch_mean | |
batch_variance | |
operation | |
reserve_space_1 | |
reserve_space_2 | |
y | |
Fungsi statis publik | |
|---|---|
DataFormat (StringPiece x) | |
Epsilon (float x) | |
IsTraining (bool x) | |
Struktur | |
|---|---|
| tensorflow::ops::FusedBatchNormV2::Attrs | Pengaturan atribut opsional untuk FusedBatchNormV2 . |
Atribut publik
rata-rata batch
::tensorflow::Output batch_mean
varians batch
::tensorflow::Output batch_variance
operasi
Operation operation
reserved_space_1
::tensorflow::Output reserve_space_1
reserved_space_2
::tensorflow::Output reserve_space_2
y
::tensorflow::Output y
Fungsi publik
FusedBatchNormV2
FusedBatchNormV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNormV2
FusedBatchNormV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV2::Attrs & attrs )
Fungsi statis publik
Format Data
Attrs DataFormat( StringPiece x )
Epsilon
Attrs Epsilon( float x )
Apakah Pelatihan
Attrs IsTraining( bool x )