jaxmat.tensors package

Submodules

• jaxmat.tensors.generic_tensors module
  • Tensor
    • Tensor.dim
    • Tensor.rank
    • Tensor.shape
    • Tensor.tensor
    • Tensor.T
    • Tensor.array
    • Tensor.array_shape
    • Tensor.rotate()
  • Tensor2
    • Tensor2.dim
    • Tensor2.rank
    • Tensor2.identity()
    • Tensor2.sym
    • Tensor2.inv
    • Tensor2.eigenvalues
    • Tensor2.T
  • SymmetricTensor2
    • SymmetricTensor2.array_shape
    • SymmetricTensor2.is_symmetric()
  • symmetric_kelvin_mandel_index_map()
  • SymmetricTensor4
    • SymmetricTensor4.dim
    • SymmetricTensor4.rank
    • SymmetricTensor4.identity()
    • SymmetricTensor4.J()
    • SymmetricTensor4.K()
    • SymmetricTensor4.array_shape
    • SymmetricTensor4.is_symmetric()
    • SymmetricTensor4.inv
  • IsotropicTensor4
    • IsotropicTensor4.kappa
    • IsotropicTensor4.mu
    • IsotropicTensor4.basis
    • IsotropicTensor4.coeffs
    • IsotropicTensor4.eval()
    • IsotropicTensor4.inv
• jaxmat.tensors.linear_algebra module
  • dim()
  • tr()
  • dev()
  • det33()
  • inv33()
  • principal_invariants()
  • main_invariants()
  • pq_invariants()
  • eig33()
  • sqrtm()
  • inv_sqrtm()
  • isotropic_function()
  • expm()
  • logm()
  • powm()
• jaxmat.tensors.tensor_utils module
  • polar()
  • sym()
  • skew()
  • axl()
  • tr()
  • dev()
  • dev_vect()
  • stretch_tensor()
  • eigenvalues_jvp()
• jaxmat.tensors.utils module
  • safe_fun()
  • safe_sqrt()
  • safe_norm()
  • FischerBurmeister()
  • rotation_matrix_direct()
  • euler_to_rotation()

Module contents