텐서플로우:: 작전:: 융합BatchNormV2

#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 텐서로 경사도 계산에 재사용됩니다.

생성자와 소멸자

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)

공개 속성

batch_mean
batch_variance
operation
reserve_space_1
reserve_space_2
y

공개 정적 함수

DataFormat (StringPiece x)
Epsilon (float x)
IsTraining (bool x)

구조체

텐서플로우:: ops:: FusedBatchNormV2:: Attrs

FusedBatchNormV2 에 대한 선택적 속성 설정자입니다.

공개 속성

배치_평균

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

공공 기능

융합BatchNormV2

 FusedBatchNormV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance
)

융합BatchNormV2

 FusedBatchNormV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance,
  const FusedBatchNormV2::Attrs & attrs
)

공개 정적 함수

데이터형식

Attrs DataFormat(
  StringPiece x
)

엡실론

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)