将pytorch转换为coreml，实现元素级最大操作

Question

我尝试使用基于coremltools的元素最大化操作将PyTorch模型转换为coreml

有了torch.max的操作，我得到了

ValueError: node input.2 (max) got 2 input(s), expected [3]

最大限度地使用火炬

RuntimeError: PyTorch convert function for op 'maximum' not implemented.

有什么办法解决这个问题吗

Answer 1

我在将pytorch元素操作转换为coreml模型时遇到了相同的问题，但通过为带有火炬前端的MIL转换器添加对torch.maximum和torch.minimum的支持，解决了这个问题

@register_torch_op
def maximum(context, node):
    inputs = _get_inputs(context, node)
    x = inputs[0]
    y = inputs[1]
    out = mb.maximum(x=x, y=y, name=node.name)
    context.add(out)