Can I write my own layer in CoreML? If possible, how? The layer I want is similar to the select_index layer in pytorch (http://pytorch.org/docs/master/torch.html#torch.index_select).
Can I write my own layer in CoreML?
Add a Comment