คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

TridiagonalSolve

ประชาชน TridiagonalSolve ชั้นสุดท้าย

แก้ระบบสมการสามมิติของสมการ

แก้ระบบสมการสามมิติของสมการ รองรับขนาดแบทช์และด้านขวาหลายด้านต่อด้านซ้ายแต่ละด้าน สำหรับ CPU โซลูชันจะคำนวณผ่านการคัดออกแบบเกาส์เซียนโดยมีหรือไม่มีการหมุนบางส่วน ทั้งนี้ขึ้นอยู่กับแอตทริบิวต์ `partial_pivoting` บน GPU ไลบรารี cuSPARSE ของ Nvidia ถูกใช้: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv แบ็กเอนด์ XLA ยังไม่รองรับการหมุนบางส่วน

คลาสที่ซ้อนกัน

ระดับ TridiagonalSolve.Options คุณลักษณะที่จำเป็นสำหรับการ TridiagonalSolve

วิธีการสาธารณะ

เอาท์พุท <T>
asOutput ()
ส่งกลับค่าแฮนเดิลเชิงสัญลักษณ์ของเทนเซอร์
คง <T> TridiagonalSolve <T>
สร้าง ( ขอบเขต ขอบเขต Operand <T> เส้นทแยงมุม, Operand <T> RHS, ตัวเลือก ... ตัวเลือก)
เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TridiagonalSolve ใหม่
เอาท์พุท <T>
เอาท์พุท ()
เทนเซอร์ของรูปร่าง `[..., M, K]` ที่มีสารละลาย
คง TridiagonalSolve.Options
partialPivoting (บูลีน partialPivoting)
คง TridiagonalSolve.Options
perturbSingular (บูลีน perturbSingular)

วิธีการสืบทอด

วิธีการสาธารณะ

ประชาชน เอาท์พุท <T> asOutput ()

ส่งกลับค่าแฮนเดิลเชิงสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการทำงานของ TensorFlow เป็นเอาต์พุตของการดำเนินการอื่นของ TensorFlow วิธีนี้ใช้เพื่อรับหมายเลขอ้างอิงเชิงสัญลักษณ์ที่แสดงถึงการคำนวณของอินพุต

สาธารณะคง TridiagonalSolve <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> เส้นทแยงมุม, Operand <T> RHS, ตัวเลือก ... ตัวเลือก)

เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TridiagonalSolve ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
เส้นทแยงมุม เทนเซอร์ของรูปร่าง `[..., 3, M]` ซึ่ง 2 มิติในสุดแสดงถึงเมทริกซ์สามเหลี่ยมที่มีแถวสามแถวเป็นเส้นเฉียงเหนือ เส้นทแยงมุม และเส้นทแยงมุมย่อยตามลำดับ องค์ประกอบสุดท้ายของ superdiagonal และองค์ประกอบแรกของ subdiagonal จะถูกละเว้น
rhs เทนเซอร์ของรูปร่าง `[..., M, K]` ซึ่งแทนค่า K ทางด้านขวามือต่อแต่ละด้านซ้ายมือ
ตัวเลือก ดำเนินการค่าแอตทริบิวต์ทางเลือก
คืนสินค้า
  • ตัวอย่างใหม่ของ TridiagonalSolve

ประชาชน เอาท์พุท <T> เอาท์พุท ()

เทนเซอร์ของรูปร่าง `[..., M, K]` ที่มีสารละลาย

สาธารณะคง TridiagonalSolve.Options partialPivoting (บูลีน partialPivoting)

พารามิเตอร์
การหมุนบางส่วน ว่าจะใช้การหมุนบางส่วนหรือไม่ การหมุนบางส่วนทำให้ขั้นตอนมีเสถียรภาพมากขึ้น แต่ช้าลง

สาธารณะคง TridiagonalSolve.Options perturbSingular (บูลีน perturbSingular)