tch-rs
tch-rs copied to clipboard
parallel map
is it possible to run map parallel for tensor? I want to apply func for each tensor with shape [x, y] in tensor with shape [n, x, y]