3 ms·
I found that there still some limitations with functorch's vmap, but I can't recall what it was.
by PartiallyTyped 4y ago
I found that there still some limitations with functorch's vmap, but I can't recall what it was.
- carbocation 4y agoProbably not your issue, but one kind of annoying bit is that the inputs need to be tensors. I ended up calling partial on the function I was messing around with and then vmapping the partial, which seemed to work.
- PartiallyTyped 4y agoPretty similar actually, I needed to pass in some parameters as tuples, lists, etc. Partial would have probably worked, but tbh I wasn't in the mood to try harder at 2am.