dneprDroid/pytorch-coreml-custom-layer-example

PyTorch to CoreML: Writing custom layers with Metal shaders - torch.nn.functional.grid_sample operation

语言：Swift · GitHub 星标：16 · Fork：2

话题：coreml · coremltools · ios · metal · ml · pytorch · swift

在 GitHub 上查看原仓库