treeflow.tf_util package

treeflow.tf_util.DEFAULT_FLOAT_DTYPE_NP

alias of float64

treeflow.tf_util.float_constant(x: float | ndarray | Iterable[float])

Converts a floating point value or array to a constant Tensor with TreeFlow’s default data type

Parameters:

x – Value that can be converted to a Tensor

Returns:

Value converted to a constant tensor

Return type:

tf.Tensor

class treeflow.tf_util.AttrsLengthMixin

Bases: object

Submodules