dimakura / SwanKit

Deep learning framework in Swift (& Metal)
MIT License
0 stars 0 forks source link

SWKTensor operations #20

Open dimakura opened 6 years ago

dimakura commented 6 years ago

Implement basic tensor operations.

Almost all operations are returning copies of tensors and have in-place counterpart. If main operation name is operationName, then mutating operation is named operationName$. Usually in-place counterparts have return type Void.

General operations

Math operations

Scalar operations

Comparison

Tensor operations

dimakura commented 6 years ago

Some references for SIMD interface: