(X)
| 49 | |
| 50 | |
| 51 | def rearrange(X): |
| 52 | # input is (32, 32, 3, N) |
| 53 | # output is (N, 3, 32, 32) |
| 54 | # N = X.shape[-1] |
| 55 | # out = np.zeros((N, 3, 32, 32), dtype=np.float32) |
| 56 | # for i in range(N): |
| 57 | # for j in range(3): |
| 58 | # out[i, j, :, :] = X[:, :, j, i] |
| 59 | # return out / 255 |
| 60 | return (X.transpose(3, 2, 0, 1) / 255).astype(np.float32) |
| 61 | |
| 62 | |
| 63 | def main(): |