a Юм>> from sympy import MatrixSymbol, Trace, eye >>> A = MatrixSymbol('A', 3, 3) >>> Trace(A) Trace(A) >>> Trace(eye(3)) Trace(Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]])) >>> Trace(eye(3)).simplify() 3 TcCs8t|Г}|jstdt|ГГВ|js,tdГВtа||бS)Nz#input to Trace, %s, is not a matrixzTrace of a non-square matrix)rZ is_Matrix┌ TypeError┌strZ is_squarer r┌__new__)┌cls┌matйr·p/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/matrices/expressions/trace.pyr"sz Trace.__new__cCs|SйNrй┌selfrrr┌_eval_transpose-szTrace._eval_transposecCsRddlm}ddlm}t||Гr2|а|ба|бS|аб}t|tГrHt В|а |бS)Nrй┌Sumщ)┌ MatrixElement)┌sympy.concrete.summationsrZmatexprr┌ isinstanceZrewrite┌diff┌doitr┌NotImplementedError┌_eval_derivative)r┌vrr┌exprrrrr 0s zTrace._eval_derivativecCs╝ddlm}m}|jdа|б}|D]Т}|jdkr`t|t||jd|jdgГdg|jdН|_n,t|t||jd|jd|jgГddgГ|_t j t j g|_|j|_|j|_d|_ d|_q$|S)Nr)┌ArrayTensorProduct┌ArrayContractionr)rщ)Z validator)rщ)Z0sympy.tensor.array.expressions.array_expressionsr#r$┌args┌_eval_derivative_matrix_linesZhigherr┌_lines┌ _validaterZOneZ_first_pointer_parentZ_second_pointer_parentZ_first_pointer_indexZ_second_pointer_index)r┌xr#r$┌r┌lrrrrr(;sD ■■° Ї ¤■ў■z#Trace._eval_derivative_matrix_linescCs |jdS)Nr)r'rrrr┌argesz Trace.argc Ksn|аddбrJ|jjfi|дО}z |абWSttfyFt|ГYS0n t|jtГr`t |jГSt|jГSdS)N┌deepT) ┌getr.rZ_eval_trace┌AttributeErrorrrrr ┌trace)r┌kwargsr.rrrris z Trace.doitcCst|jабГабSr)rr.┌as_explicitrrrrrr4wszTrace.as_explicitcs┤ddlm}ddlmЙ|jЙtИ|Гr░ЗЗfddД}tttИj ГГ|dН}tИj |ИГrЖИИГа бЙtttИj ГГЗfddДdН}|аИj |dЕИj d|ЕбЙtИГS|S) Nr)┌MatMul)┌ Transposecs"Иj|}t|ИГr|j}t|ГSr)r'rr.r)r+┌aйr6┌ trace_argrr┌get_arg_keyГs z%Trace._normalize..get_arg_key)┌keycstИj|ГSr)rr')r+)r9rr┌Мєz"Trace._normalize..) Z!sympy.matrices.expressions.matmulr5Z$sympy.matrices.expressions.transposer6r.r┌min┌range┌lenr'rZfromiterr)rr5r:Zindminrr8r┌ _normalizezs "zTrace._normalizecKs:ddlm}tdГ}||j||f|d|jjdfГабS)Nrr┌ir)rrrr.┌rowsr)rr"r3rrBrrr┌_eval_rewrite_as_SumСszTrace._eval_rewrite_as_SumN)┌__name__┌ __module__┌__qualname__┌__doc__Zis_TraceZis_commutativerrr r(┌propertyr.rr4rArDrrrrrs* rcCst|ГабS)aTrace of a Matrix. Sum of the diagonal elements. Examples ======== >>> from sympy import trace, Symbol, MatrixSymbol, eye >>> n = Symbol('n') >>> X = MatrixSymbol('X', n, n) # A square matrix >>> trace(2*X) 2*Trace(X) >>> trace(eye(3)) 3 )rr)r"rrrr2Чsr2N)Zsympy.core.basicrZsympy.core.exprrrZsympy.core.singletonrZsympy.core.sortingrZsympy.core.symbolrZsympy.core.sympifyrZsympy.matrices.matricesr Zsympy.matrices.commonr rr2rrrr┌s