Skip to content

Instantly share code, notes, and snippets.

@Huijun-Cui
Huijun-Cui / pad_packed_demo.py
Created June 22, 2018 03:28 — forked from Tushar-N/pad_packed_demo.py
How to use pad_packed_sequence in pytorch
import torch
import torch.nn as nn
from torch.autograd import Variable
from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
import torch.nn.functional as F
import numpy as np
import itertools
def flatten(l):
return list(itertools.chain.from_iterable(l))