Module dopt.core.ops.math

Contains common maths operations.

Functions

NameDescription
argmin(input, axis, mod, line)Performs an argmin over the specified dimension.
matmul(lhs, rhs, mod, line)Computes the matrix multiplication between two rank-2 tensors.
maxElement(op, axes, mod, line)Computes a max reduction along the specified axes.
sum(op, axes, mod, line)Computes a sum reduction along the specified axes.