элегантный способ сделать обратную матрицу в тензорном потоке

Question

Я бы хотел инвертировать кучу тензоров в списке, используя разложение по Холесскому в tenorflow 2, но полученный код довольно уродлив. есть ли какой-нибудь элегантный / более pythoni c способ сделать что-то вроде этого:

iMps = []
for Mp in Mps :
    cholMp  = tf.linalg.cholesky(Mp)
    icholMp = tf.linalg.inv(cholMp)
    iMp = tf.tensordot(tf.transpose(icholMp),icholMp)
    iMps.append(iMp)

можно ли заменить l oop другим материалом?, Mps - список тензоров с другим размером ( я могу представить это как что-то еще?). Есть ли способ сделать его более элегантным?