pad_tensor

pyjet.layers.pad_tensor(tensor, length, pad_value=0.0, dim=0)

---

pad_sequences

pyjet.layers.pad_sequences(tensors, pad_value=0.0, length_last=False)

---

unpad_sequences

pyjet.layers.unpad_sequences(padded_tensors, seq_lens, length_last=False)

---

pack_sequences

pyjet.layers.pack_sequences(tensors)

---

unpack_sequences

pyjet.layers.unpack_sequences(packed_tensors, seq_lens)

---

kmax_pooling

pyjet.layers.kmax_pooling(x, dim, k)

---

pad_numpy_to_length

pyjet.layers.pad_numpy_to_length(x, length)

---

seq_softmax

pyjet.layers.seq_softmax(x, return_padded=False)