Math Operations Parallel
Located in Gradien.Ops.Math and Gradien.Ops.BLAS. These operations are parallelized.
Element-wise
All element-wise operations support broadcasting semantics implicitly if implemented in the kernel.
.add
lua
(A: Tensor, B: Tensor) -> Tensor.sub
lua
(A: Tensor, B: Tensor) -> Tensor.mul
lua
(A: Tensor, B: Tensor) -> Tensor.scalarAdd
Adds a scalar value s to every element of A.
lua
(A: Tensor, s: number) -> Tensor.scalarMul
Multiplies every element of A by scalar s.
lua
(A: Tensor, s: number) -> TensorLinear Algebra
.matmul
Performs Matrix Multiplication.
lua
(A: Tensor, B: Tensor) -> Tensor