treeflow.tree.topology.tensorflow_tree_topology module

treeflow.tree.topology.tensorflow_tree_topology.tensor_taxon_count(parent_indices: Tensor) → Tensor

class treeflow.tree.topology.tensorflow_tree_topology.TensorflowTreeTopology(parent_indices: Tensor, child_indices: Tensor = None, preorder_indices: Tensor = None, taxon_set: TaxonSet | None = None)

Bases: TensorflowTreeTopologyAttrs

property taxon_set: TaxonSet | None

classmethod get_event_ndims() → TensorflowTreeTopology

get_prefer_static_rank() → TensorflowTreeTopology

classmethod rank_to_has_batch_dimensions(rank: TensorflowTreeTopology)

has_batch_dimensions() → bool | Tensor

numpy() → NumpyTreeTopology

child_indices: Tensor

preorder_indices: Tensor

property node_child_indices: Tensor

parent_indices: Tensor

property postorder_node_indices: Tensor

property preorder_node_indices: Tensor

property taxon_count: Tensor

Number of leaf taxa the tree is based on (n)