Method __init__

inspiremusic/transformer/embedding.py:182–185

(self, d_model: int, dropout_rate: float)

    """

    def __init__(self, d_model: int, dropout_rate: float):
        super().__init__()
        self.d_model = d_model
        self.dropout = torch.nn.Dropout(p=dropout_rate)

    def forward(self,
                x: torch.Tensor,