dopt.core.ops.math

Contains common maths operations.

Members

Functions

argmin
Operation argmin(Operation input, size_t axis, string mod, size_t line)

Performs an argmin over the specified dimension.

initialize
void initialize()
Undocumented in source. Be warned that the author may not have intended to support it.
matmul
Operation matmul(Operation lhs, Operation rhs, string mod, size_t line)

Computes the matrix multiplication between two rank-2 tensors.

maxElement
Operation maxElement(Operation op, size_t[] axes, string mod, size_t line)

Computes a max reduction along the specified axes.

sum
Operation sum(Operation op, size_t[] axes, string mod, size_t line)

Computes a sum reduction along the specified axes.

Meta

Authors

Henry Gouk