(self, decoder_layer, num_layers, norm)
| 125 | __constants__ = ['norm'] |
| 126 | |
| 127 | def __init__(self, decoder_layer, num_layers, norm): |
| 128 | super().__init__() |
| 129 | self.layers = transformer._get_clones(decoder_layer, num_layers) |
| 130 | self.num_layers = num_layers |
| 131 | self.norm = norm |
| 132 | |
| 133 | def forward( |
| 134 | self, |