Open
Description
detail | 详细描述 | 詳細な説明
因为在yolo11n_pnnx.py中还存在这样的代码
v_93, v_94 = torch.split(tensor=v_92, dim=1, split_size_or_sections=(128,128))
v_95 = self.model_10_m_0_attn_qkv_conv(v_94)
v_96 = v_95.view(1, 2, 128, 400)
v_97, v_98, v_99 = torch.split(tensor=v_96, dim=2, split_size_or_sections=(32,32,64))
v_100 = torch.transpose(input=v_97, dim0=-2, dim1=-1)
v_101 = torch.matmul(input=v_100, other=v_98)
v_102 = (v_101 * 1.767767e-01)
v_103 = F.softmax(input=v_102, dim=-1)
v_104 = torch.transpose(input=v_103, dim0=-2, dim1=-1)
v_105 = torch.matmul(input=v_99, other=v_104)
v_106 = v_105.view(1, 128, 20, 20)
v_107 = v_99.reshape(1, 128, 20, 20)
v_108 = self.model_10_m_0_attn_pe_conv(v_107)
不知道这里该如何处理,能否麻烦有人可以帮忙看看
Metadata
Assignees
Labels
No labels
Activity