MolecularDiffusion.modules.layers.equiformer_v2.edge_rot_mat

Copyright (c) Meta Platforms, Inc. and affiliates.

Functions

init_edge_rot_mat(edge_distance_vec)

Module Contents

MolecularDiffusion.modules.layers.equiformer_v2.edge_rot_mat.init_edge_rot_mat(edge_distance_vec)