tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

parallel map

Open igor-yusupov opened this issue 1 year ago • 0 comments

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]

igor-yusupov avatar Apr 28 '23 19:04 igor-yusupov