Skip to main content
Ctrl+K

PMM

  • Models
  • Modules
  • Examples
  • Advanced
  • GitHub
  • Models
  • Modules
  • Examples
  • Advanced
  • GitHub

Section Navigation

  • conformal Module
  • typing Module
  • model_util Module
  • tree_util Module
  • eigen_util Module
  • graph_util Module
  • Advanced

Advanced#

  • conformal Module

Most of the following modules are mostly intended for internal use.

  • typing Module
    • ArrayData
    • ArrayDataShape
    • BatchlessComplexDataFixed
    • BatchlessDataFixed
    • BatchlessRealDataFixed
    • ComplexDataFixed
    • Data
    • DataFixed
    • DataShape
    • DictParams
    • DictState
    • HyperParams
    • ListParams
    • ListState
    • ModuleCallable
    • NonSerializable
    • Params
    • RealDataFixed
    • Serializable
    • State
    • TupleParams
    • TupleState
  • model_util Module
    • ModelCallable
    • ModelModules
    • ModelParams
    • ModelParamsStruct
    • ModelState
    • ModelStateStruct
    • autobatch()
  • tree_util Module
    • abs()
    • abs_sqr()
    • abs_sqrt()
    • add()
    • all_close()
    • all_equal()
    • astype()
    • batch_leaves()
    • concatenate()
    • div()
    • extend_structure_from_strpaths()
    • get_shapes()
    • getitem_by_strpath()
    • has_uniform_leaf_shapes()
    • is_shape_leaf()
    • is_single_leaf()
    • make_mutable()
    • mean()
    • mul()
    • neg()
    • pow()
    • pytree_split()
    • random_permute_leaves()
    • safecast()
    • scalar_add()
    • scalar_mul()
    • setitem_by_strpath()
    • shapes_equal()
    • sqrt()
    • strfmt_pytree()
    • sub()
    • uniform_leaf_shapes_equal()
  • eigen_util Module
    • select_eigenpairs_by_eigenvalue()
    • select_eigenpairs_by_fn()
    • select_eigenvalues()
    • validate_eigensystem_input_shape()
  • graph_util Module
    • concretize_connections()
    • concretize_paths()
    • get_outer_connections_by_tree()
    • partition_connections_by_tree()
    • place_connections_in_tree()
    • resolve_connections()
    • validate_resolution()

previous

Regression Example

next

conformal Module

This Page

  • Show Source