dagma.nonlinear.DagmaMLP.h_func
¶
-
dagma.nonlinear.DagmaMLP.h_func(s: float =
1.0
) torch.Tensor ¶ Constrain 2-norm-squared of fc1 weights along m1 dim to be a DAG
Last update:
Jan 14, 2024
dagma.nonlinear.DagmaMLP.h_func
¶1.0
) torch.Tensor ¶Constrain 2-norm-squared of fc1 weights along m1 dim to be a DAG