:param mlp: list of int :param bn: whether to use batchnorm
(self, *, mlp: List[int], bn: bool = True)
| 117 | r"""Propigates the features of one set to another""" |
| 118 | |
| 119 | def __init__(self, *, mlp: List[int], bn: bool = True): |
| 120 | """ |
| 121 | :param mlp: list of int |
| 122 | :param bn: whether to use batchnorm |
| 123 | """ |
| 124 | super().__init__() |
| 125 | self.mlp = pt_utils.SharedMLP(mlp, bn=bn) |
| 126 | |
| 127 | def forward( |
| 128 | self, unknown: torch.Tensor, known: torch.Tensor, unknow_feats: torch.Tensor, known_feats: torch.Tensor |