Toybrick»开源社区 › 技术论坛 › 人工智能 › ReduceSum算子编译失败

ReduceSum算子编译失败

gef1998

新手上路

积分: 4

楼主

发表于 3 天前 查看: 82|回复: 0 | [复制链接]

| 只看该作者

onnx如下：

报错信息如下：

对应torch的代码为：

mask_mems = (torch.abs(feat_mems) > 0).float() #feat_mems的shape为[1, 6, 128, 50, 8, 50]
denom = torch.sum(mask_mems, dim=1, keepdim=False)

本主题由 toybrick 于 3 天前审核通过

收藏0

使用道具举报

返回列表