(X)
| 37 | |
| 38 | |
| 39 | def rearrange(X): |
| 40 | # input is (32, 32, 3, N) |
| 41 | # output is (N, 32, 32, 3) |
| 42 | # N = X.shape[-1] |
| 43 | # out = np.zeros((N, 32, 32, 3), dtype=np.float32) |
| 44 | # for i in xrange(N): |
| 45 | # for j in xrange(3): |
| 46 | # out[i, :, :, j] = X[:, :, j, i] |
| 47 | # return out / 255 |
| 48 | return (X.transpose(3, 0, 1, 2) / 255).astype(np.float32) |
| 49 | |
| 50 | |
| 51 | def main(): |