shape = [len(seqs)]
// Find the maximum dimension along each axis. That"s what we"ll pad to.
dim_sizes = zip(*[seq.shape for seq in seqs])
shape.extend(max(sizes) for sizes in dim_sizes)
// Now copy the data into our new buffer.
output: Array = self.alloc(tuple(shape), dtype=seqs[0].dtype)
for i, arr in enumerate(seqs):
// TODO: It would be nice to generalize this to work along different
After Change
if len(set(seq.shape[1:] for seq in seqs)) != 1:
raise ValueError("Cannot pad sequences that differ on other dimensions")
// Find the maximum dimension along each axis. That"s what we"ll pad to.
shapes = [tuple(s.shape) for s in seqs]
dim_sizes = zip(*[shape for shape in shapes])
max_dims = [max(sizes) for sizes in dim_sizes]final_shape = (len(seqs),) + tuple(max_dims)
output = self.alloc(final_shape, dtype=seqs[0].dtype)
for i, arr in enumerate(seqs):
region = [i] + [slice(0, dim) for dim in arr.shape]
output[region] = arr
