Dimenet 操作符#

径向基函数#

dimenet.radial_basis_fwd

计算Dimenet++径向基特征的前向传递。

dimenet.radial_basis_bwd

计算Dimenet++径向基特征的反向传播。

dimenet.radial_basis_bwd_bwd

计算Dimenet++径向基特征的第二阶反向传播。

边缘到边缘聚合#

dimenet.agg_edge_to_edge_fwd

计算Dimenet++交互块聚合层的前向传递。

dimenet.agg_edge_to_edge_bwd

计算Dimenet++交互块聚合层的反向传播。

dimenet.agg_edge_to_edge_bwd2_grad

计算Dimenet++交互块的二阶反向传播

dimenet.agg_edge_to_edge_bwd2_main

计算Dimenet++交互块的二阶反向传递