dense
Dense layer implementation for ProteinMPNN.
-
prxteinmpnn.model.dense.dense_layer(layer_parameters, node_features)[source]
Apply a dense layer to node features.
- Return type:
Int[Array, 'num_atoms num_features']
- Parameters:
layer_parameters (PyTree[str, 'P'])
node_features (Int[Array, 'num_atoms num_features'])