Functions — github.com/apple/ml-probabilistic-attention

CodeHub MCPcopy Create free account

hub / github.com/apple/ml-probabilistic-attention / functions

Functions3 in github.com/apple/ml-probabilistic-attention

⨍ Functions 3 ◇ Types & classes 1

(self, uniform_query_precision=False, uniform_value_precision=False, magnitude_priors=False,

probabilisticattention.py:12

Runs an update of the probabilistic version of attention based on a Mixture of Gaussians model. This layer is equivalent to a standar

probabilisticattention.py:22

Methodupdate_weights

probabilisticattention.py:55