(self, repeats)
| 107 | : repeats: int, the number of repetitions for the vector. |
| 108 | """ |
| 109 | def __init__(self, repeats): |
| 110 | super(RepeatVector, self).__init__() |
| 111 | self.repeats = repeats |
| 112 | |
| 113 | def forward(self, x): |
| 114 | x = x.unsqueeze(-1).repeat(1, 1, self.repeats) # <------------ Mejorar? |