Hi,
I’m afraid there is no map in pytorch.
If all the operations are very small, single threaded CPU will be the fastest I’m afraid.
If you can share your problem, maybe we can help you achieve some parallelization using the existing functions though.