- 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.
Contains common maths operations.