สร้าง TensorArray สำหรับจัดเก็บการไล่ระดับสีของค่าในแฮนเดิลที่กำหนด
หากมีการไล่ระดับ TensorArray ที่ระบุอยู่แล้ว ให้ส่งคืนการอ้างอิง
ล็อคขนาดของ TensorArray ดั้งเดิมโดยปิดใช้งานแฟล็กขนาดไดนามิก
**หมายเหตุเกี่ยวกับอินพุต flow_in:**
หมายเลขอ้างอิง flow_in บังคับให้การดำเนินการค้นหาการไล่ระดับสีเกิดขึ้นหลังจากการดำเนินการอื่นบางอย่างเกิดขึ้นเท่านั้น ตัวอย่างเช่น เมื่อฟอร์เวิร์ด TensorArray มีขนาดแบบไดนามิก การเขียนไปยัง TensorArray นี้อาจปรับขนาดวัตถุ TensorArray แบบไล่ระดับจะมีขนาดคงที่ตามขนาดของ TensorArray ไปข้างหน้าเมื่อการดำเนินการนี้ดำเนินการ นอกจากนี้ ขนาดของฟอร์เวิร์ด TensorArray จะถูกตรึงไว้โดยการโทรนี้ ด้วยเหตุนี้ โฟลว์จึงถูกใช้เพื่อให้แน่ใจว่าการเรียกเพื่อสร้าง Gradient TensorArray จะเกิดขึ้นหลังจากดำเนินการเขียนทั้งหมดแล้วเท่านั้น
ในกรณีของ TensorArray ที่มีขนาดไดนามิก การคำนวณเกรเดียนต์ควรทำเฉพาะในการดำเนินการอ่านที่ผูกมัดตัวเองผ่านโฟลว์ที่จะเกิดขึ้นหลังจากดำเนินการเขียนทั้งหมดแล้วเท่านั้น ด้วยวิธีนี้จะทราบขนาดสุดท้ายของ TensorArray ไปข้างหน้าเมื่อมีการเรียกการดำเนินการนี้
**หมายเหตุเกี่ยวกับแอตทริบิวต์ต้นทาง:**
การเรียกไล่ระดับ TensorArray ใช้วัตถุ TensorArray แบบสะสม หากมีการคำนวณและเรียกใช้การไล่ระดับสีหลายรายการในเซสชันเดียวกัน โหนดการไล่ระดับสีหลายโหนดอาจไหลผ่าน TensorArray ตัวสะสมเดียวกันโดยไม่ได้ตั้งใจ การนับสองครั้งนี้และโดยทั่วไปจะทำลายโฟลว์การไล่ระดับสี TensorArray
วิธีแก้ไขคือการระบุว่าการไล่ระดับสีใดที่เรียกการไล่ระดับสี TensorArray นี้โดยเฉพาะ ซึ่งดำเนินการโดยระบุสตริงที่ไม่ซ้ำกัน (เช่น "การไล่ระดับสี", "gradients_1", ...) จากชื่อของเทนเซอร์การไล่ระดับสีอินพุต สตริงนี้ใช้เป็นคำต่อท้ายเมื่อสร้างวัตถุไล่ระดับ TensorArray ที่นี่ (แอตทริบิวต์ `แหล่งที่มา')
แอตทริบิวต์ `แหล่งที่มา' ถูกเพิ่มเป็นส่วนต่อท้ายชื่อของ TensorArray ไปข้างหน้าเมื่อดำเนินการสร้าง / ค้นหา เพื่อให้การคำนวณการไล่ระดับสีแยกกันได้รับตัวสะสม TensorArray ของตัวเอง
วิธีการสาธารณะ
คง TensorArrayGrad | สร้าง ( ขอบเขต ขอบเขต Operand <?> จับ ตัวถูกดำเนินการ <ลอย> flowin แหล่ง String) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TensorArrayGrad ใหม่ |
เอาท์พุท <ลอย> | flowOut () |
เอาท์พุท <?> | gradHandle () |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง TensorArrayGrad สร้าง ( ขอบเขต ขอบเขต Operand <?> จับ ตัวถูกดำเนินการ <ลอย> flowin แหล่ง String)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TensorArrayGrad ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
รับมือ | ที่จับสำหรับไปข้างหน้า TensorArray |
ไหลเข้า | สเกลาร์ทุ่นที่บังคับใช้การโยงการดำเนินการอย่างเหมาะสม |
แหล่งที่มา | สตริงซอร์สการไล่ระดับสี ใช้เพื่อตัดสินใจว่า TensorArray สำหรับการไล่ระดับสีใดที่จะส่งคืน |
คืนสินค้า
- ตัวอย่างใหม่ของ TensorArrayGrad