การทำให้เป็นมาตรฐานแบบแบตช์
โปรดทราบว่าขนาดของเทนเซอร์ 4D ถูกกำหนดโดย "NHWC" หรือ "NCHW" ขนาดของเทนเซอร์ 1D ตรงกับมิติ C ของเทนเซอร์ 4D
คลาสที่ซ้อนกัน
| ระดับ | FusedBatchNorm ตัวเลือก | แอ็ตทริบิวต์ทางเลือกสำหรับ FusedBatchNorm | |
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| เอาท์พุต <U> | ค่าเฉลี่ยแบทช์ () 1D Tensor สำหรับค่าเฉลี่ยแบทช์ที่คำนวณ ซึ่ง TensorFlow ใช้เพื่อคำนวณค่าเฉลี่ยที่ทำงานอยู่ |
| เอาท์พุต <U> | ชุดผลต่าง () 1D Tensor สำหรับความแปรปรวนแบบแบตช์ที่คำนวณแล้ว ที่จะใช้โดย TensorFlow เพื่อคำนวณความแปรปรวนที่ทำงานอยู่ |
| คงที่ <T ขยาย TNumber , U ขยาย TNumber > FusedBatchNorm <T, U> | สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> x, ตัวดำเนินการ <U> สเกล, ตัวดำเนินการ <U> ออฟเซ็ต, ตัวถูกดำเนินการ <U> ค่าเฉลี่ย, ตัวดำเนินการ <U> ความแปรปรวน, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ FusedBatchNorm ใหม่ |
| FusedBatchNorm.Options แบบคงที่ | dataFormat (สตริง dataFormat) |
| FusedBatchNorm.Options แบบคงที่ | เอปไซลอน (เอปไซลอนลอย) |
| FusedBatchNorm.Options แบบคงที่ | เอ็กซ์โปเนนเชียลAvgFactor (โฟลตเอ็กซ์โปเนนเชียลAvgFactor) |
| FusedBatchNorm.Options แบบคงที่ | isTraining (บูลีน isTraining) |
| เอาท์พุต <U> | สำรอง Space1 () เทนเซอร์ 1D สำหรับค่าเฉลี่ยแบทช์ที่คำนวณได้ เพื่อนำมาใช้ซ้ำในการคำนวณเกรเดียนต์ |
| เอาท์พุต <U> | สำรอง Space2 () เทนเซอร์ 1D สำหรับความแปรปรวนแบทช์ที่คำนวณแล้ว (ความแปรปรวนแบบกลับด้านในกรณี cuDNN) เพื่อนำมาใช้ซ้ำในการคำนวณการไล่ระดับสี |
| เอาท์พุต <U> | สำรอง Space3 () เทนเซอร์ 1D สำหรับผลลัพธ์ระดับกลางบางส่วน เพื่อนำมาใช้ซ้ำในการคำนวณการไล่ระดับสีเพื่อประสิทธิภาพที่ดีขึ้น |
| เอาท์พุต <T> | ใช่ () เทนเซอร์ 4D สำหรับข้อมูลเอาท์พุต |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
เอาท์พุท สาธารณะ <U> matchMean ()
1D Tensor สำหรับค่าเฉลี่ยแบทช์ที่คำนวณ ซึ่ง TensorFlow ใช้เพื่อคำนวณค่าเฉลี่ยที่ทำงานอยู่
เอาท์พุท สาธารณะ <U> ชุดความแปรปรวน ()
1D Tensor สำหรับความแปรปรวนแบบแบตช์ที่คำนวณแล้ว ที่จะใช้โดย TensorFlow เพื่อคำนวณความแปรปรวนที่ทำงานอยู่
สร้าง FusedBatchNorm <T, U> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> x, ตัวดำเนินการ <U> สเกล, ตัวดำเนิน การ <U> ออฟเซ็ต, ตัวดำเนินการ <U> ค่าเฉลี่ย, ตัวดำเนินการ <U> ความแปรปรวน, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ FusedBatchNorm ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| x | เทนเซอร์ 4D สำหรับป้อนข้อมูล |
| มาตราส่วน | เทนเซอร์ 1D สำหรับปัจจัยการปรับขนาด เพื่อปรับขนาด x ที่ทำให้เป็นมาตรฐาน |
| ชดเชย | เทนเซอร์ 1D สำหรับออฟเซ็ต เพื่อเลื่อนไปที่ x ที่ทำให้เป็นมาตรฐาน |
| หมายถึง | 1D Tensor สำหรับค่าเฉลี่ยประชากร ใช้สำหรับการอนุมานเท่านั้น จะต้องว่างเปล่าสำหรับการฝึกอบรม |
| ความแปรปรวน | เทนเซอร์ 1D สำหรับความแปรปรวนของประชากร ใช้สำหรับการอนุมานเท่านั้น จะต้องว่างเปล่าสำหรับการฝึกอบรม |
| ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ FusedBatchNorm
สาธารณะ FusedBatchNorm.Options dataFormat แบบคงที่ (String dataFormat)
พารามิเตอร์
| ข้อมูลรูปแบบ | รูปแบบข้อมูลสำหรับ x และ y "NHWC" (ค่าเริ่มต้น) หรือ "NCHW" |
|---|
epsilon FusedBatchNorm.Options สาธารณะแบบคงที่ (epsilon ลอย)
พารามิเตอร์
| เอปไซลอน | จำนวนทศนิยมเล็กๆ ที่บวกเข้ากับความแปรปรวนของ x |
|---|
สาธารณะ FusedBatchNorm.Options isTraining แบบคงที่ (บูลีน isTraining)
พารามิเตอร์
| คือการฝึกอบรม | ค่าบูลเพื่อระบุการดำเนินการมีไว้สำหรับการฝึก (ค่าเริ่มต้น) หรือการอนุมาน |
|---|
เอาท์พุท สาธารณะ <U> ReserveSpace1 ()
เทนเซอร์ 1D สำหรับค่าเฉลี่ยแบทช์ที่คำนวณได้ เพื่อนำมาใช้ซ้ำในการคำนวณเกรเดียนต์
เอาท์พุท สาธารณะ <U> ReserveSpace2 ()
เทนเซอร์ 1D สำหรับความแปรปรวนแบทช์ที่คำนวณแล้ว (ความแปรปรวนแบบกลับด้านในกรณี cuDNN) เพื่อนำมาใช้ซ้ำในการคำนวณการไล่ระดับสี
เอาท์พุท สาธารณะ <U> ReserveSpace3 ()
เทนเซอร์ 1D สำหรับผลลัพธ์ระดับกลางบางส่วน เพื่อนำมาใช้ซ้ำในการคำนวณการไล่ระดับสีเพื่อประสิทธิภาพที่ดีขึ้น