I’d better explain my request with an example.
let there be a sequence of integers
s = [2, 1, 3]. I want to create a mask from this sequence as follows
mask = [ [1, 1, 0, 0, 0, 0], [0, 0, 1, 0, 0, 0], [0, 0, 0, 1, 1, 1] ]. each row of 2-d mask contains
s[i] ones, and these ones going successive. thus, len of each row of
Moreover, I want to effectively complete a task with batched input (a batch of such sequences). ofc, rows of masks in a batch would not be equal, so I want to pad it with trailing zeros.
I have quite a weird vectorized code for the task, but I want it to be simpler. If anyone interested, I’ll provide a code in comment below.