TridiagonalSolve

TridiagonalSolve คลาสสุดท้ายสาธารณะ

แก้ระบบสมการตรีโกณมิติ

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

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

ระดับ TridiagonalSolve.Options คุณลักษณะทางเลือกสำหรับ TridiagonalSolve

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

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

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

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

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "TridiagonalSolve"

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

เอาท์ พุท สาธารณะ <T> asOutput ()

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

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

สร้าง TridiagonalSolve <T> แบบคงที่สาธารณะ ( ขอบเขตขอบเขต ตัวดำเนินการ <T> เส้นทแยงมุม ตัวดำเนินการ <T> rhs ตัวเลือก... ตัวเลือก)

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

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

เอาท์พุท สาธารณะ <T> เอาท์พุท ()

เทนเซอร์ของรูปร่าง `[..., M, K]` ที่มีคำตอบ

TridiagonalSolve.Options แบบคงที่สาธารณะ ( Boolean PartialPivoting)

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