텐서플로우:: 작전:: 퓨즈배치표준
#include <nn_ops.h>
일괄 정규화.
요약
4D 텐서의 크기는 "NHWC" 또는 "NCHW"로 정의됩니다. 1D 텐서의 크기는 4D 텐서의 차원 C와 일치합니다.
인수:
- 범위: 범위 개체
- x: 입력 데이터를 위한 4D 텐서 .
- scale: 정규화된 x의 크기를 조정하기 위한 배율 인수용 1D 텐서입니다 .
- offset: 정규화된 x로 이동하기 위한 오프셋용 1D 텐서 입니다.
- 평균: 모집단 평균에 대한 1D 텐서입니다 . 추론에만 사용됩니다. 훈련을 위해서는 비어 있어야 합니다.
- variance: 모집단 분산을 위한 1D 텐서입니다 . 추론에만 사용됩니다. 훈련을 위해서는 비어 있어야 합니다.
선택적 속성( Attrs
참조):
- 엡실론: x의 분산에 추가되는 작은 부동소수점 숫자입니다.
- data_format: x와 y의 데이터 형식입니다. "NHWC"(기본값) 또는 "NCHW"입니다.
- is_training: 작업이 훈련(기본값) 또는 추론을 위한 것임을 나타내는 부울 값입니다.
보고:
-
Output
y: 출력 데이터를 위한 4D 텐서 . -
Output
배치_평균: 계산된 배치 평균에 대한 1D 텐서로 , 텐서플로우에서 실행 평균을 계산하는 데 사용됩니다. -
Output
배치_변수: 계산된 배치 분산을 위한 1D 텐서로 , TensorFlow에서 실행 분산을 계산하는 데 사용됩니다. -
Output
Reserve_space_1: 계산된 배치 평균을 위한 1D 텐서로 기울기 계산에 재사용됩니다. -
Output
Reserve_space_2: 계산된 배치 분산(cuDNN의 경우 반전 분산)을 위한 1D 텐서로 경사도 계산에 재사용됩니다.
생성자와 소멸자 | |
---|---|
FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance) | |
FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance, const FusedBatchNorm::Attrs & attrs) |
공개 속성 | |
---|---|
batch_mean | |
batch_variance | |
operation | |
reserve_space_1 | |
reserve_space_2 | |
y |
공개 정적 함수 | |
---|---|
DataFormat (StringPiece x) | |
Epsilon (float x) | |
ExponentialAvgFactor (float x) | |
IsTraining (bool x) |
구조체 | |
---|---|
tensorflow:: ops:: FusedBatchNorm:: Attrs | FusedBatchNorm 에 대한 선택적 속성 설정자입니다. |
공개 속성
배치_평균
::tensorflow::Output batch_mean
배치_분산
::tensorflow::Output batch_variance
작업
Operation operation
예비_공간_1
::tensorflow::Output reserve_space_1
예비_공간_2
::tensorflow::Output reserve_space_2
와이
::tensorflow::Output y
공공 기능
퓨즈배치표준
FusedBatchNorm( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
퓨즈배치표준
FusedBatchNorm( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNorm::Attrs & attrs )
공개 정적 함수
데이터형식
Attrs DataFormat( StringPiece x )
엡실론
Attrs Epsilon( float x )
지수평균 인자
Attrs ExponentialAvgFactor( float x )
IsTraining
Attrs IsTraining( bool x )