nifty8.re.tree_math package# Submodules# nifty8.re.tree_math.forest_math module assert_arithmetics() get_map() has_arithmetics() map_forest() map_forest_mean() mean() mean_and_std() random_like() stack() tree_shape() unite() unstack() nifty8.re.tree_math.pytree_string module PyTreeString PyTreeString.__init__() PyTreeString.lower() PyTreeString.startswith() PyTreeString.tree_flatten() PyTreeString.tree_unflatten() PyTreeString.upper() hide_strings() nifty8.re.tree_math.vector module Vector Vector.__init__() Vector.conj() Vector.conjugate() Vector.copy() Vector.dot() Vector.imag Vector.max() Vector.min() Vector.ravel() Vector.real Vector.shape Vector.size Vector.sum() Vector.tree Vector.tree_flatten() Vector.tree_unflatten() nifty8.re.tree_math.vector_math module ShapeWithDtype ShapeWithDtype.__init__() ShapeWithDtype.dtype ShapeWithDtype.from_leave() ShapeWithDtype.ndim ShapeWithDtype.shape ShapeWithDtype.size all() any() conj() conjugate() dot() matmul() max() min() norm() result_type() shape() size() sum() vdot() where() Module contents#