مؤتمر Google I / O هو التفاف! تابع جلسات TensorFlow اعرض الجلسات

تينسورفلو :: العمليات :: فيوزد باتشنورم

#include <nn_ops.h>

تطبيع الدفعة.

ملخص

لاحظ أن حجم Tensors 4D يتم تحديده بواسطة "NHWC" أو "NCHW". يتطابق حجم Tensors 1D مع البعد C للموتر 4D.

الحجج:

  • النطاق: كائن النطاق
  • x: موتر رباعي الأبعاد لبيانات الإدخال.
  • المقياس: موتر 1D لعامل القياس ، لقياس x المقيس.
  • الإزاحة: موتر 1D للإزاحة ، للتحول إلى x المقيس.
  • يعني: موتر 1D لوسط السكان. تستخدم للاستدلال فقط ؛ يجب أن تكون فارغة للتدريب.
  • التباين: موتر 1D للتباين السكاني. تستخدم للاستدلال فقط ؛ يجب أن تكون فارغة للتدريب.

السمات الاختيارية (انظر Attrs ):

  • إبسيلون: رقم عائم صغير يضاف إلى تباين x.
  • data_format: تنسيق البيانات لـ x و y. إما "NHWC" (افتراضي) أو "NCHW".
  • is_training: قيمة منطقية للإشارة إلى أن العملية مخصصة للتدريب (افتراضي) أو للاستدلال.

عائدات:

  • Output y: موتر 4D لبيانات الإخراج.
  • Output batch_mean: A 1D التنسور للمتوسط دفعة محسوبة، لاستخدامها من قبل TensorFlow لحساب متوسط قيد التشغيل.
  • Output batch_variance: Tensor 1D لتباين الدُفعة المحسوب ، ليتم استخدامه بواسطة TensorFlow لحساب التباين الجاري.
  • Output reserve_space_1: موتر أحادي الأبعاد لوسط الدُفعة المحسوب ، ليتم إعادة استخدامه في حساب التدرج.
  • Output reserve_space_2: A 1D التنسور لهذه الفروق دفعة المحوسب (التباين مقلوب في حالة cuDNN)، لإعادة استخدامها في حساب التدرج.

البنائين والمدمرين

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)
IsTraining (bool x)

الهياكل

tensorflow :: العمليات :: 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 IsTraining(
  bool x
)