org.apache.clojure-mxnet.executor
arg-arrays
(arg-arrays executor)
arg-map
(arg-map executor)
backward
(backward executor)(backward executor ndarray-or-vec)
* Do backward pass to get the gradient of arguments.
* @param ndarray-or-vec Gradient on the outputs to be propagated back.
* This parameter is only needed when bind is called
* on outputs that are not a loss function.
forward
(forward executor)(forward executor is-train kwargs)
* Calculate the outputs specified by the binded symbol.
* @param is-train whether this forward is for evaluation purpose.
* @param kwargs Additional specification of input arguments.
get-grad
(get-grad executor grad-name)
grad-arrays
(grad-arrays executor)
grad-map
(grad-map executor)
outputs
(outputs executor)
reshape
(reshape executor kwargs {:keys [partial-shaping allow-up-sizing], :or {partial-shaping false, allow-up-sizing false}})(reshape executor kwargs)
* Return a new executor with the same symbol and shared memory,
* but different input/output shapes.
* For runtime reshaping, variable length sequences, etc.
* The returned executor shares state with the current one,
* and cannot be used in parallel with it.
* @param kwargs Map of string to shape-vec.
* - new shape for arguments.
* @parms opts with :partial-shaping Whether to allow changing the shape of unspecified arguments.
* and :allow-up-sizing Whether to allow allocating new ndarrays that's larger than the original.
set-arg
(set-arg executor arg-name arg-val-or-vec)
set-arg-arrays
(set-arg-arrays executor vec-of-ndarray-or-val)