(self, dims_in)
| 4 | |
| 5 | class LocalDynamics(nn.Module): |
| 6 | def __init__(self, dims_in): |
| 7 | |
| 8 | super(LocalDynamics, self).__init__() |
| 9 | self.linear = nn.Conv1d(dims_in*2, dims_in, kernel_size=1, stride=1, padding=0) |
| 10 | self.softmax = nn.Softmax(dim = -1) |
| 11 | self.k = 1 |
| 12 | def forward(self, x, mask): |
| 13 | mask = F.interpolate(mask.detach(), size=x.size()[2:], mode='nearest') |
| 14 | B, C, H, W = x.shape |
nothing calls this directly
no outgoing calls
no test coverage detected