MCPcopy
hub / github.com/ml-jku/hopfield-layers / get_projected_pattern_matrix

Method get_projected_pattern_matrix

hflayers/__init__.py:257–272  ·  view source on GitHub ↗

Fetch Hopfield projected pattern matrix gathered by passing through the specified data. :param input: data to be passed through the Hopfield association :param stored_pattern_padding_mask: mask to be applied on stored patterns :param association_mask: mask to be app

(self, input: Union[Tensor, Tuple[Tensor, Tensor, Tensor]],
                                     stored_pattern_padding_mask: Optional[Tensor] = None,
                                     association_mask: Optional[Tensor] = None)

Source from the content-addressed store, hash-verified

255 association_mask=association_mask)[2]
256
257 def get_projected_pattern_matrix(self, input: Union[Tensor, Tuple[Tensor, Tensor, Tensor]],
258 stored_pattern_padding_mask: Optional[Tensor] = None,
259 association_mask: Optional[Tensor] = None) -> Tensor:
260 """
261 Fetch Hopfield projected pattern matrix gathered by passing through the specified data.
262
263 :param input: data to be passed through the Hopfield association
264 :param stored_pattern_padding_mask: mask to be applied on stored patterns
265 :param association_mask: mask to be applied on inner association matrix
266 :return: pattern projection matrix as computed by the Hopfield core module
267 """
268 with torch.no_grad():
269 return self._associate(
270 data=input, return_projected_patterns=True,
271 stored_pattern_padding_mask=stored_pattern_padding_mask,
272 association_mask=association_mask)[3]
273
274 @property
275 def batch_first(self) -> bool:

Calls 1

_associateMethod · 0.95

Tested by

no test coverage detected