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)
| 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: |
no test coverage detected