Cholesky

คำนวณการสลายตัวของ Cholesky ของเมทริกซ์จตุรัสตั้งแต่หนึ่งเมทริกซ์ขึ้นไป

ข้อมูลเข้าเป็นเทนเซอร์ของรูปร่าง `[..., M, M]` ซึ่งมี 2 มิติด้านในสุดเป็นเมทริกซ์จตุรัส

ข้อมูลเข้าจะต้องมีความสมมาตรและเป็นค่าบวกแน่นอน เฉพาะส่วนสามเหลี่ยมด้านล่างของอินพุตเท่านั้นที่จะถูกนำมาใช้สำหรับการดำเนินการนี้ ส่วนสามเหลี่ยมด้านบนจะไม่ถูกอ่าน

เอาต์พุตเป็นเทนเซอร์ที่มีรูปร่างเดียวกันกับอินพุตที่มีการสลายตัวของ Cholesky สำหรับเมทริกซ์ย่อยอินพุตทั้งหมด `[..., :, :]`

หมายเหตุ : การคำนวณการไล่ระดับสีบน GPU จะเร็วกว่าสำหรับเมทริกซ์ขนาดใหญ่ แต่ไม่ใช่สำหรับขนาดแบตช์ขนาดใหญ่เมื่อเมทริกซ์ย่อยมีขนาดเล็ก ในกรณีนี้ การใช้ CPU อาจเร็วกว่า