我有这样的模式：https://github.com/williamyang1991/DualStyleGAN，并尝试将其转换为CoreML。到目前为止，我创建了原始Colab笔记本的副本，并在末尾添加了两个块：

!pip install coremltools
import coremltools as ct

和

# @title Convert inverted image.

inverted_latent = torch.Tensor(result_latents[0][4]).cuda().unsqueeze(0).unsqueeze(1)

with torch.no_grad():
    net.eval()

    [sampled_src, sampled_dst] = net(inverted_latent, input_is_latent=True)[0]

    traced_model = torch.jit.trace(net, inverted_latent)
    mlmodel = ct.convert(traced_model, inputs=[ct.ImageType(name="input", shape=inverted_latent.shape,bias=[-1,-1,-1],scale=2.0/255.0)])
    mlmodel.save("modelsaved.mlmodel")

要运行它，您应该将任何带有face的图像放到/content中，并在/usr/local/lib/python3.7/dist-packages/torchvision/transforms/functional.py中将第545行的round方法替换为第546行的np.round
但它在
mlmodel = ct.转换（...
与：
运行时错误：未实现操作“pythonop”的PyTorch转换函数。
我建议用可以被转换的方法重写这个模块，对吗？但是我不知道如何找到这个模块的源代码。
所以我的问题是：如果我想的方法正确，我怎样才能找到模块的来源？如果我想的方法错误，请告诉我正确的方法。