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)结果：
