Sqrtm

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

มัทมัล(sqrtm(A), sqrtm(A)) = A

เมทริกซ์อินพุตควรกลับด้านได้ หากเมทริกซ์อินพุตเป็นจำนวนจริง ก็ไม่ควรมีค่าลักษณะเฉพาะที่เป็นจำนวนจริงและเป็นค่าลบ (อนุญาตให้ใช้คู่ค่าลักษณะเฉพาะของคอนจูเกตที่ซับซ้อนได้)

รากที่สองของเมทริกซ์คำนวณโดยการลดขนาดเมทริกซ์ให้เหลือรูปแบบกึ่งสามเหลี่ยมด้วยการสลายตัวของ Schur จริง จากนั้นจึงคำนวณรากที่สองของเมทริกซ์กึ่งสามเหลี่ยมโดยตรง รายละเอียดของอัลกอริทึมมีอยู่ใน: Nicholas J. Higham, "การคำนวณรากที่สองที่แท้จริงของเมทริกซ์จริง", Linear Algebra Appl., 1987

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