Deep Learning, Sora, Computer Vision, Transformers, Attention.
everyimage has different sequence length, lets pad all the sequences to same length. we can use torch rnn utils pad_sequence for this from torch.nn.utils.rnn import pad_sequence pout = pad_sequence([i.squeeze(0) for i in out], batch_first=True) pout.shape #torch.Size([4, 480, 768]) this is the