Initialize SinusoidalPositionEncoder. Args: d_model: D Model instance. dropout_rate: TODO.
(self, d_model=80, dropout_rate=0.1)
| 26 | """ """ |
| 27 | |
| 28 | def __init__(self, d_model=80, dropout_rate=0.1): |
| 29 | """Initialize SinusoidalPositionEncoder. |
| 30 | |
| 31 | Args: |
| 32 | d_model: D Model instance. |
| 33 | dropout_rate: TODO. |
| 34 | """ |
| 35 | super().__init__() |
| 36 | |
| 37 | def encode( |
| 38 | self, positions: torch.Tensor = None, depth: int = None, dtype: torch.dtype = torch.float32 |