jaxmat.tensors.tensor_utils module# polar(F, mode='RU')[source]# Computes the ‘RU’ or ‘VR’ polar decomposition of F. sym(A)[source]# Computes the symmetric part of a tensor. skew(A)[source]# Computes the skew part of a tensor. axl(A)[source]# Computes the axial part of a skew symmetric tensor. If not skew-symmetric, this function takes the skew-symmetric part first. tr(A)[source]# Trace of a n-dim 2nd-rank tensor. dev(A)[source]# dev_vect(A)[source]# stretch_tensor(F)[source]# Computes the strech tensor U = sqrtm(F.T @ F). eigenvalues_jvp(primals, tangents)[source]#
