Table Of Contents
Table Of Contents

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) –