mx.infer.rnn.one.unroll

Description

Inference for one-to-one unroll models

Usage

mx.infer.rnn.one.unroll(

  infer.data,

  symbol,

  num_hidden,

  arg.params,

  aux.params,

  init_states = NULL,

  ctx = mx.cpu()

)

Arguments

Argument

Description

infer.data

NDArray

symbol

Model used for inference

num_hidden

ctx