openfold.model.outer_product_mean¶

Classes

OuterProductMean(c_m, c_z, c_hidden[, eps])

Implements Algorithm 10.

class OuterProductMean(c_m, c_z, c_hidden, eps=0.001)¶

Bases: Module

Implements Algorithm 10.

forward(m, mask=None, chunk_size=None, inplace_safe=False)¶

Parameters:

m (Tensor)
mask (Tensor | None)
chunk_size (int | None)
inplace_safe (bool)

Return type: