y = x.transpose(0, 2, 3, 1) ... split in small patches and transpose each patch rearrange(ims, 'b (h1 h2) (w1 w2) c ... tensor(320., dtype=torch.float64).
確定! 回上一頁