liblaf.apple.warp.math

Functions:

• cw_square –
• fro_norm_square –

  \(\norm{M}_F^2\).

• polar_rv –
• square –
• svd_rv –

cw_square

cw_square(a: Any) -> Any

Source code in src/liblaf/apple/warp/math/_misc.py

@wp.func
def cw_square(a: Any) -> Any:
    return wp.cw_mul(a, a)

fro_norm_square

fro_norm_square(M: Any) -> Any

\(\norm{M}_F^2\).

Source code in src/liblaf/apple/warp/math/_misc.py

@wp.func
def fro_norm_square(M: Any) -> Any:
    r"""$\norm{M}_F^2$."""
    return wp.ddot(M, M)

polar_rv

polar_rv(A: mat33) -> tuple[mat33, mat33]

Source code in src/liblaf/apple/warp/math/_rotation.py

@wp.func
@no_type_check
def polar_rv(A: mat33) -> tuple[mat33, mat33]:
    U, s, V = svd_rv(A)
    R = U @ wp.transpose(V)
    S = V @ wp.diag(s) @ wp.transpose(V)
    return R, S

square

square(a: Any) -> Any

Source code in src/liblaf/apple/warp/math/_misc.py

@wp.func
def square(a: Any) -> Any:
    return a * a

svd_rv

svd_rv(A: mat33) -> tuple[mat33, vec3, mat33]

Source code in src/liblaf/apple/warp/math/_rotation.py

@wp.func
@no_type_check
def svd_rv(A: mat33) -> tuple[mat33, vec3, mat33]:
    U, s, V = wp.svd3(A)
    return U, s, V