:param x: tuple (s, V) of `torch.Tensor`, or single `torch.Tensor` (will be assumed to be scalar channels)
(self, x)
| 188 | self.vdropout = _VDropout(drop_rate) |
| 189 | |
| 190 | def forward(self, x): |
| 191 | ''' |
| 192 | :param x: tuple (s, V) of `torch.Tensor`, |
| 193 | or single `torch.Tensor` |
| 194 | (will be assumed to be scalar channels) |
| 195 | ''' |
| 196 | if type(x) is torch.Tensor: |
| 197 | return self.sdropout(x) |
| 198 | s, v = x |
| 199 | return self.sdropout(s), self.vdropout(v) |
| 200 | |
| 201 | |
| 202 | class GVPLayerNorm(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected