Deep Learning, Sora, Computer Vision, Transformers, Attention.
w, patch_size) y = torch.arange(0, h, patch_size) X, Y = torch.meshgrid(x, y) centers = torch.stack((X.flatten(), Y.flatten()), dim=1) centers = centers+(patch_size/2) centers.shape #torch.Size([432, 2]) fig, ax = plt.subplots(figsize=(2.76, 3.68)) ax.scatter(centers[:, 0].numpy(), centers[:,