Normalizacja wsadowa.
Należy zauważyć, że rozmiar tensorów 4D jest zdefiniowany przez „NHWC” lub „NCHW”. Rozmiar tensorów 1D odpowiada wymiarowi C tensorów 4D.
Klasy zagnieżdżone
klasa | Opcje FusedBatchNorm | Opcjonalne atrybuty dla FusedBatchNorm |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <U> | partiaŚrednia () Tensor 1D dla obliczonej średniej wsadowej, który ma być używany przez TensorFlow do obliczania średniej bieżącej. |
Wyjście <U> | wariancja partii () Tensor 1D dla obliczonej wariancji wsadowej, który ma być używany przez TensorFlow do obliczania bieżącej wariancji. |
statyczny <T rozszerza TNumber , U rozszerza TNumber > FusedBatchNorm <T, U> | |
statyczne FusedBatchNorm.Options | dataFormat (Format danych ciągu) |
statyczne FusedBatchNorm.Options | epsilon (pływający epsilon) |
statyczne FusedBatchNorm.Options | wykładniczyAvgFactor (Float wykładniczyAvgFactor) |
statyczne FusedBatchNorm.Options | isTraining (wartość logiczna isTraining) |
Wyjście <U> | rezerwaPrzestrzeń1 () Tensor 1D dla obliczonej średniej wsadowej do ponownego wykorzystania w obliczeniach gradientu. |
Wyjście <U> | rezerwaPrzestrzeń2 () Tensor 1D dla obliczonej wariancji wsadowej (odwrócona wariancja w przypadku cuDNN), do ponownego wykorzystania w obliczeniach gradientu. |
Wyjście <U> | rezerwaPrzestrzeń3 () Tensor 1D dla niektórych wyników pośrednich, który można ponownie wykorzystać w obliczeniach gradientu w celu uzyskania lepszej wydajności. |
Wyjście <T> | y () Tensor 4D dla danych wyjściowych. |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <U> partiaŚrednia ()
Tensor 1D dla obliczonej średniej wsadowej, który ma być używany przez TensorFlow do obliczania średniej bieżącej.
publiczne dane wyjściowe <U> wsadowe ()
Tensor 1D dla obliczonej wariancji wsadowej, który ma być używany przez TensorFlow do obliczania bieżącej wariancji.
public static FusedBatchNorm <T, U> create ( zakres zakresu , argument <T> x, skala argumentu <U>, przesunięcie argumentu <U>, średnia argumentu <U>, wariancja argumentu <U>, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację FusedBatchNorm.
Parametry
zakres | aktualny zakres |
---|---|
X | Tensor 4D dla danych wejściowych. |
skala | Tensor 1D dla współczynnika skalowania do skalowania znormalizowanego x. |
zrównoważyć | Tensor 1D dla przesunięcia, aby przesunąć do znormalizowanego x. |
mieć na myśli | Tensor 1D dla średniej populacji. Używane wyłącznie do wnioskowania; musi być pusty na potrzeby treningu. |
zmienność | Tensor 1D dla wariancji populacji. Używane wyłącznie do wnioskowania; musi być pusty na potrzeby treningu. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja FusedBatchNorm
publiczny statyczny FusedBatchNorm.Options dataFormat (String dataFormat)
Parametry
Format danych | Format danych dla x i y. Albo „NHWC” (domyślnie) albo „NCHW”. |
---|
publiczny statyczny FusedBatchNorm.Options epsilon (Float epsilon)
Parametry
epsilon | Mała liczba zmiennoprzecinkowa dodana do wariancji x. |
---|
public static FusedBatchNorm.Options isTraining (Boolean isTraining)
Parametry
jestSzkolenie | Wartość bool wskazująca, że operacja służy do uczenia (domyślnie) lub wnioskowania. |
---|
publiczne wyjście <U> ReserveSpace1 ()
Tensor 1D dla obliczonej średniej wsadowej do ponownego wykorzystania w obliczeniach gradientu.
publiczne wyjście <U> ReserveSpace2 ()
Tensor 1D dla obliczonej wariancji wsadowej (odwrócona wariancja w przypadku cuDNN), do ponownego wykorzystania w obliczeniach gradientu.
publiczne wyjście <U> ReserveSpace3 ()
Tensor 1D dla niektórych wyników pośrednich, który można ponownie wykorzystać w obliczeniach gradientu w celu uzyskania lepszej wydajności.