rearrange

rearrange

一、head=1

1.

代码

import numpy as np
import torch
from einops import rearrange
data = torch.from_numpy(np.random.rand(1, 6, 3, 2))
print(str(data))
data = rearrange(data, 'b (head c) h w -> b head c (h w)', head=1)
print(data)
print(data.shape)

结果

标题

2.

代码：

import numpy as np
import torch
from einops import rearrange
data = torch.from_numpy(np.random.rand(2, 3, 3, 2))
print(str(data))
data = rearrange(data, 'b (head c) h w -> b head c (h w)', head=1)
print(data)
print(data.shape)

结果：

二、head=2

3.

代码：

import numpy as np
import torch
from einops import rearrange
data = torch.from_numpy(np.random.rand(2, 4, 3, 2))
print(str(data))
data = rearrange(data, 'b (head c) h w -> b head c (h w)', head=2)
print(data)
print(data.shape)

结果：

4.

代码：

import numpy as np
import torch
from einops import rearrange
data = torch.from_numpy(np.random.rand(2, 6, 3, 2))
print(str(data))
data = rearrange(data, 'b (head c) h w -> b head c (h w)', head=2)
print(data)
print(data.shape)

结果：