Crate custos_math

source ·

Re-exports

pub use raw_ops::*;
pub use cpu::*;
pub use cuda::*;
pub use opencl::*;

Modules

cpu
cuda
custos
matrix_multiply
nn
opencl
raw_ops
raw_prelude

Structs

Matrix
A matrix using Buffer described with rows and columns

Traits

AdditionalOps
AssignOps
Assignment operations
BaseOps
Element-wise +, -, *, / operations for matrices.
ClipOp
ColOp
CudaTranspose
DiagflatOp
FnsOps
Gemm
Matrix multiplication. Uses provided device.
MaxOps
RandBuf
RandOp
RowOp
ScalarAssign
SumOps
SumOverOps
TransposeOp

Functions

cl_to_cpu_lr
OpenCL
cl_to_cpu_s
OpenCL
cu_clip
diagflat
ew_op
rand_slice
slice_transpose