Kernel Operations on GPU or CPU, with Autodiff, without Memory Overflows

[Up] [Top]

Documentation for package ‘rkeops’ version 2.1.0

Help Pages

A B C E G I K L M N O P R S T U V W X misc

rkeops-package rkeops

-- A --

abs Element-wise absolute value.
acos Element-wise arc-cosine.
Arg Element-wise angle (or argument) of complex.
argKmin argKmin.
argKmin_reduction argKmin reduction.
argmax ArgMax.
argmax_reduction ArgMax reduction.
argmin ArgMin.
argmin_reduction Argmin reduction.
asin Element-wise arc-sine.
atan Element-wise arc-tangent.
atan2 Element-wise 2-argument arc-tangent.

-- B --

binaryop.LazyTensor Build a binary operation

-- C --

check_rkeops Check if 'rkeops' is ready and working
clamp Element-wise clamp function.
clampint Element-wise clampint function.
clean_rkeops Clean RKeOps build directory
concat Concatenation.
Conj Element-wise complex conjugate.
cos Element-wise cosine.

-- E --

elem Elem.
elemT ElemT.
exp Element-wise exponential.
exp1j Element-wise "complex exponential of 1j x" operation.
extract Extract.
extractT ExtractT.

-- G --

get_rkeops_options Get the current 'rkeops' options in 'R' global options scope
grad Gradient operation.

-- I --

ifelse Element-wise if-else function.
Im Element-wise imaginary part of complex.
imag2complex Element-wise "imaginary 2 complex" operation.
inv Element-wise 1/x inverse.
is.ComplexLazyParameter is.ComplexLazyParameter?
is.ComplexLazyTensor is.ComplexLazyTensor?
is.int Scalar integer test.
is.LazyMatrix is.LazyMatrix?
is.LazyParameter is.LazyParameter?
is.LazyTensor is.LazyTensor?
is.LazyVector is.LazyVector?

-- K --

keops_grad Compute the gradient of a rkeops operator
keops_kernel Defines a new KeOps operator to compute a specific symbolic formula
Kmin Kmin.
Kmin_argKmin Kmin-argKmin.
Kmin_argKmin_reduction Kmin-argKmin reduction.
Kmin_reduction Kmin reduction.

-- L --

LazyTensor Build and return a LazyTensor object.
log Element-wise natural logarithm.
logsumexp Log-Sum-Exp reduction.
logsumexp_reduction Log-Sum-Exp reduction.

-- M --

matvecmult Matrix-vector product.
max Maximum.
max.default Maximum.
max_argmax Max-ArgMax.
max_argmax_reduction Max-ArgMax reduction.
max_reduction Max reduction.
min Minimum.
min.default Minimum.
min_argmin Min-ArgMin.
min_argmin_reduction Min-ArgMin reduction.
min_reduction Min reduction.
Mod Element-wise absolute value (or modulus).
mod Element-wise modulo with offset function.

-- N --

norm2 Euclidean norm.
normalize Vector normalization.

-- O --

one_hot One-hot.

-- P --

Pm Wrapper LazyTensor parameter.

-- R --

Re Element-wise real part of complex.
real2complex Element-wise "real 2 complex" operation.
reduction.LazyTensor Reduction operation.
relu Element-wise ReLU function.
rkeops rkeops
rkeops_disable_verbosity Disable additional verbosity in 'rkeops'
rkeops_enable_verbosity Enable additional verbosity in 'rkeops'
rkeops_use_cpu Enable GPU-computing when calling user-defined operators
rkeops_use_float32 Use 32bit float precision in computations
rkeops_use_float64 Use 64bit float precision in computations
rkeops_use_gpu Enable GPU-computing when calling user-defined operators
round Element-wise rounding function.
rsqrt Element-wise inverse square root.

-- S --

set_rkeops_options Initialize or update 'rkeops' options in 'R' global options scope
sign Element-wise sign.
sin Element-wise sine.
sinxdivx Element-wise sin(x)/x function.
sqdist Squared distance.
sqnorm2 Squared Euclidean norm.
sqrt Element-wise square root.
square Element-wise square.
step Element-wise step function.
sum Sum operation or Sum reduction.
sum.default Sum operation.
sumsoftmaxweight Sum of weighted Soft-Max reduction.
sumsoftmaxweight_reduction Sum of weighted Soft-Max reduction.
sum_reduction Sum reduction.

-- T --

tensorprod Tensor product.
ternaryop.LazyTensor Build a ternary operation

-- U --

unaryop.LazyTensor Build a unary operation

-- V --

vecmatmult Vector-matrix product.
Vi Wrapper LazyTensor indexed by "i".
Vj Wrapper LazyTensor indexed by "j".

-- W --

weightedsqdist Generic squared distance.
weightedsqnorm Generic squared Euclidean norm.

-- X --

xlogx Element-wise x*log(x) function.

-- misc --

%*% Matrix multiplication.
* Multiplication.
+ Addition.
/ Division.
^ Power.
| Euclidean scalar product.