Deep Learning, Sora, Computer Vision, Transformers, Attention.
torch.repeat_interleave(mask.unsqueeze(-1), 480, dim=2)#.unsqueeze(1) mask_.shape #torch.Size([4, 480, 480]) fig, ax = plt.subplots(figsize=(4, 3), nrows=2, ncols=2) for m_ in range(mask_.shape[0]): ax.flat[m_].imshow(np.uint8(255*(~mask_[m_].bool()).numpy())) ax.flat[m_].axis("off") plt.show()