topobench.nn.backbones.graph.nsd_utils package

Utility modules for the Neural Sheaf Diffusion (NSD) backbone.

Submodules

• topobench.nn.backbones.graph.nsd_utils.inductive_discrete_models module
  • DiagLaplacianBuilder
    • DiagLaplacianBuilder.__init__()
    • DiagLaplacianBuilder.forward()
  • GeneralLaplacianBuilder
    • GeneralLaplacianBuilder.__init__()
    • GeneralLaplacianBuilder.forward()
  • InductiveDiscreteBundleSheafDiffusion
    • InductiveDiscreteBundleSheafDiffusion.__init__()
    • InductiveDiscreteBundleSheafDiffusion.forward()
    • InductiveDiscreteBundleSheafDiffusion.get_param_size()
    • InductiveDiscreteBundleSheafDiffusion.left_right_linear()
  • InductiveDiscreteDiagSheafDiffusion
    • InductiveDiscreteDiagSheafDiffusion.__init__()
    • InductiveDiscreteDiagSheafDiffusion.forward()
  • InductiveDiscreteGeneralSheafDiffusion
    • InductiveDiscreteGeneralSheafDiffusion.__init__()
    • InductiveDiscreteGeneralSheafDiffusion.forward()
    • InductiveDiscreteGeneralSheafDiffusion.left_right_linear()
  • LocalConcatSheafLearner
    • LocalConcatSheafLearner.__init__()
    • LocalConcatSheafLearner.forward()
  • NormConnectionLaplacianBuilder
    • NormConnectionLaplacianBuilder.__init__()
    • NormConnectionLaplacianBuilder.forward()
  • SheafDiffusion
    • SheafDiffusion.__init__()
• topobench.nn.backbones.graph.nsd_utils.laplace module
  • compute_learnable_diag_laplacian_indices()
  • compute_learnable_laplacian_indices()
  • compute_left_right_map_index()
  • mergesp()
• topobench.nn.backbones.graph.nsd_utils.laplacian_builders module
  • DiagLaplacianBuilder
    • DiagLaplacianBuilder.__init__()
    • DiagLaplacianBuilder.forward()
  • GeneralLaplacianBuilder
    • GeneralLaplacianBuilder.__init__()
    • GeneralLaplacianBuilder.forward()
  • LaplacianBuilder
    • LaplacianBuilder.__init__()
    • LaplacianBuilder.scalar_normalise()
  • NormConnectionLaplacianBuilder
    • NormConnectionLaplacianBuilder.__init__()
    • NormConnectionLaplacianBuilder.forward()
  • Orthogonal
    • Orthogonal.__init__()
    • Orthogonal.forward()
  • compute_learnable_diag_laplacian_indices()
  • compute_learnable_laplacian_indices()
  • compute_left_right_map_index()
  • degree()
  • mergesp()
  • scatter_add()
• topobench.nn.backbones.graph.nsd_utils.orthogonal module
  • Orthogonal
    • Orthogonal.__init__()
    • Orthogonal.forward()
• topobench.nn.backbones.graph.nsd_utils.sheaf_base module
  • SheafDiffusion
    • SheafDiffusion.__init__()
• topobench.nn.backbones.graph.nsd_utils.sheaf_models module
  • LocalConcatSheafLearner
    • LocalConcatSheafLearner.__init__()
    • LocalConcatSheafLearner.forward()
  • SheafLearner
    • SheafLearner.__init__()
    • SheafLearner.forward()
    • SheafLearner.set_L()
  • abstractmethod()