ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: FusedBatchNormGrad

#include <nn_ops.h>

배치 정규화를위한 기울기.

요약

4D Tensor의 크기는 "NHWC"또는 "NCHW"로 정의됩니다. 1D Tensor의 크기는 4D Tensor의 C 차원과 일치합니다.

인수 :

  • 범위 : 범위 개체
  • y_backprop : y에 대한 기울기에 대한 4D 텐서 .
  • x : 입력 데이터 용 4D 텐서 .
  • 스케일 : 스케일링 인자에 대한 1D 텐서 로 정규화 된 x를 스케일링합니다.
  • reserve_space_1 : is_training이 True이면 계산 된 배치에 대한 1D Tensor 가 그래디언트 계산에 재사용됨을 의미합니다. is_training이 False이면 모집단에 대한 1D Tensor 가 1 차 및 2 차 기울기 계산 모두에서 재사용됨을 의미합니다.
  • reserve_space_2 : is_training이 True 인 경우 계산 된 배치 분산 (cuDNN 케이스의 반전 분산)에 대한 1D 텐서 가 기울기 계산에 재사용됩니다. is_training이 False이면 모집단 분산에 대한 1D 텐서 가 1 차 및 2 차 기울기 계산 모두에서 재사용됩니다.

선택적 속성 ( Attrs 참조) :

  • 엡실론 : x의 분산에 더해지는 작은 부동 숫자.
  • data_format : y_backprop, x, x_backprop의 데이터 형식입니다. "NHWC"(기본값) 또는 "NCHW"입니다.
  • is_training : 훈련 (기본값) 또는 추론을위한 작업임을 나타내는 부울 값입니다.

보고:

  • Output x_backprop : x에 대한 기울기에 대한 4D 텐서 .
  • Output scale_backprop : 스케일에 대한 기울기의 1D 텐서 .
  • Output offset_backprop : 오프셋에 대한 기울기의 1D 텐서 .
  • Output reserve_space_3 : FusedBatchNorm 의 평균 입력과 일치하는 사용되지 않은 자리 표시 자.
  • Output reserve_space_4 : FusedBatchNorm 의 분산 입력과 일치하는 사용되지 않은 자리 표시 자.

생성자와 소멸자

FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2)
FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2, const FusedBatchNormGrad::Attrs & attrs)

공용 속성

offset_backprop
operation
reserve_space_3
reserve_space_4
scale_backprop
x_backprop

공개 정적 함수

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

구조체

tensorflow :: ops :: FusedBatchNormGrad :: Attrs

FusedBatchNormGrad에 대한 선택적 속성 설정자.

공용 속성

offset_backprop

::tensorflow::Output offset_backprop

조작

Operation operation

Reserve_space_3

::tensorflow::Output reserve_space_3

Reserve_space_4

::tensorflow::Output reserve_space_4

scale_backprop

::tensorflow::Output scale_backprop

x_backprop

::tensorflow::Output x_backprop

공공 기능

FusedBatchNormGrad

 FusedBatchNormGrad(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input y_backprop,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input reserve_space_1,
  ::tensorflow::Input reserve_space_2
)

FusedBatchNormGrad

 FusedBatchNormGrad(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input y_backprop,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input reserve_space_1,
  ::tensorflow::Input reserve_space_2,
  const FusedBatchNormGrad::Attrs & attrs
)

공개 정적 함수

DataFormat

Attrs DataFormat(
  StringPiece x
)

엡실론

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)