x1 = [8,32,256,256]
x1 = x1.permute((0, 2, 3, 1)).flatten(1, 2)
x1.permute((0, 2, 3, 1)): [8,32,256,256]—>[8,256,256,32] 次元転送
flatten(1, 2): [8,256,256,32] —>[8,65536,32] 2. 3 次元データは 1 次元に償却され、4 次元データは 3 次元データに縮小されます。
x1 = [8,32,256,256]
x1 = x1.permute((0, 2, 3, 1)).flatten(1, 2)
x1.permute((0, 2, 3, 1)): [8,32,256,256]—>[8,256,256,32] 次元転送
flatten(1, 2): [8,256,256,32] —>[8,65536,32] 2. 3 次元データは 1 次元に償却され、4 次元データは 3 次元データに縮小されます。