(self, intensors, N)
| 178 | class DataCollatorWithPadding: |
| 179 | |
| 180 | def paddingtensor(self, intensors, N): |
| 181 | B, n, S = intensors.shape |
| 182 | # padding_tensor = torch.zeros(B, N - n, S,dtype=intensors.dtype) |
| 183 | padding_tensor = torch.zeros(B, N - n, S) |
| 184 | outtensors = torch.cat((intensors, padding_tensor), dim=1) |
| 185 | return outtensors |
| 186 | |
| 187 | def paddingtensor2D(self, intensors, N): |
| 188 | B, n = intensors.shape |