แก้ระบบสมการตรีโกณมิติ
แก้ระบบสมการตรีโกณมิติ รองรับขนาดแบทช์และด้านขวาหลายรายการต่อด้านซ้ายแต่ละด้าน บน 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
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง TridiagonalSolve <T> แบบคงที่สาธารณะ ( ขอบเขตขอบเขต ตัวดำเนินการ <T> เส้นทแยงมุม ตัวดำเนินการ <T> rhs ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TridiagonalSolve ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
เส้นทแยงมุม | เทนเซอร์ของรูปร่าง `[..., 3, M]` ซึ่งมี 2 มิติด้านในสุดแทนเมทริกซ์ตรีทแยงมุม โดยมี 3 แถวคือเส้นทแยงมุมเหนือ เส้นทแยงมุม และเส้นทแยงมุมย่อย ตามลำดับ องค์ประกอบสุดท้ายของเส้นทแยงมุมเหนือและองค์ประกอบแรกของเส้นทแยงมุมย่อยจะถูกละเว้น |
Rhs | เทนเซอร์ของรูปร่าง `[..., M, K]` แทน K ทางด้านขวามือแต่ละข้างทางซ้าย |
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- ตัวอย่างใหม่ของ TridiagonalSolve
TridiagonalSolve.Options แบบคงที่สาธารณะ ( Boolean PartialPivoting)
พารามิเตอร์
การหมุนบางส่วน | ไม่ว่าจะใช้การหมุนรอบบางส่วนหรือไม่ การหมุนบางส่วนทำให้ขั้นตอนมีเสถียรภาพมากขึ้น แต่ช้าลง |
---|