(self, intensors, N)
| 169 | class DataCollatorWithPadding: |
| 170 | |
| 171 | def paddingtensor(self, intensors, N): |
| 172 | B, n, S = intensors.shape |
| 173 | # padding_tensor = torch.zeros(B, N - n, S,dtype=intensors.dtype) |
| 174 | padding_tensor = torch.zeros(B, N - n, S) |
| 175 | outtensors = torch.cat((intensors, padding_tensor), dim=1) |
| 176 | return outtensors |
| 177 | |
| 178 | def paddingtensor2D(self, intensors, N): |
| 179 | B, n = intensors.shape |