Table Of Contents
Table Of Contents

mxnet.autograd.mark_variables

mxnet.autograd.mark_variables(variables, gradients, grad_reqs='write')[source]

Mark NDArrays as variables to compute gradient for autograd.

Parameters
  • variables (NDArray or list of NDArray) –

  • gradients (NDArray or list of NDArray) –

  • grad_reqs (str or list of str) –