dopt.core.ops.math

Contains common maths operations.

Members

Functions

argmin
Operation argmin(Operation input, size_t axis, string mod = __MODULE__, size_t line = __LINE__)

Performs an argmin over the specified dimension.

matmul
Operation matmul(Operation lhs, Operation rhs, string mod = __MODULE__, size_t line = __LINE__)

Computes the matrix multiplication between two rank-2 tensors.

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

Computes a max reduction along the specified axes.

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

Computes a sum reduction along the specified axes.

Meta

Authors

Henry Gouk