Parallel
Namespace: Tensor
Parent Module: Tensor
multi-threaded tensor operations
Functions and values
| Function or value | Description |
foldAxis fn initial axis a
Signature: fn:('T -> 'TA -> 'T) -> initial:'T -> axis:int -> a:Tensor<'TA> -> Tensor<'T>
Type parameters: 'T, 'TA |
folds the function over the given axis |
init dev shape fn
Signature: dev:ITensorDevice -> shape:int64 list -> fn:(int64 [] -> 'T) -> Tensor<'T>
Type parameters: 'T |
creates a new tensor with the values returned by the function. |
map fn a
Signature: fn:('T -> 'R) -> a:Tensor<'T> -> Tensor<'R>
Type parameters: 'T, 'R |
maps all elements using the specified function into a new tensor |
map2 fn a b
Signature: fn:('TA -> 'TB -> 'R) -> a:Tensor<'TA> -> b:Tensor<'TB> -> Tensor<'R>
Type parameters: 'TA, 'TB, 'R |
maps all elements using the specified function into a new tensor |
mapi fn a
Signature: fn:(int64 [] -> 'T -> 'R) -> a:Tensor<'T> -> Tensor<'R>
Type parameters: 'T, 'R |
maps all elements using the specified indexed function into a new tensor |
mapi2 fn a b
Signature: fn:(int64 [] -> 'TA -> 'TB -> 'R) -> a:Tensor<'TA> -> b:Tensor<'TB> -> Tensor<'R>
Type parameters: 'TA, 'TB, 'R |
maps all elements using the specified indexed function into a new tensor |
