Method
__init__
(self, model: nn.Module, typ: str, N, T=1, bipartition=None)
Source from the content-addressed store, hash-verified
| 76 | n=N -> t=T |
| 77 | """ |
| 78 | def __init__(self, model: nn.Module, typ: str, N, T=1, bipartition=None): |
| 79 | super().__init__(model) |
| 80 | self.typ = typ |
| 81 | self.T = T |
| 82 | self.N = N |
| 83 | self.bipartition = bipartition or '_' in self.typ |
| 84 | logging.info('DTCTWrapper with typ={}, T={}, N={}, bipartition={}'.format(typ, T, N, bipartition)) |
| 85 | |
| 86 | def __call__(self, xn, n): |
| 87 | if np.isscalar(n): |
Callers
nothing calls this directly
Tested by
no test coverage detected