def masked_seq(seq_i):
seq_i_len=data.seq_len(seq_i)
new_seq_i=np.zeros((max_seq,seq_dim))
new_seq_i[:seq_i_len]=seq_i[:seq_i_len]
return new_seq_i
def make_mask(x,n_batch,max_seq):
mask = np.zeros((n_batch, max_seq),dtype=float)
for i,seq_i in enumerate(x):
seq_i_len=data.seq_len(seq_i)
mask[i][:seq_i_len]=1.0
return mask
