treeflow.evolution.substitution.eigendecomposition module

class treeflow.evolution.substitution.eigendecomposition.Eigendecomposition(eigenvectors: Tensor, inverse_eigenvectors: Tensor, eigenvalues: Tensor)

Bases: object

Eigendecomposition of an instantaneous rate matrix

eigenvectors

2D Tensor with right eigenvectors as columns

Type:

tensorflow.python.framework.tensor.Tensor

inverse_eigenvectors

2D Tensor, inverse of eigenvectors

Type:

tensorflow.python.framework.tensor.Tensor

eigenvalues

1D Tensor of eigenvalues

Type:

tensorflow.python.framework.tensor.Tensor

eigenvectors: Tensor
inverse_eigenvectors: Tensor
eigenvalues: Tensor
add_inner_batch_dimensions(batch_dims: int = 1, inner_batch_rank: int = 0) Eigendecomposition

Add batch dimensions before the state dimensions