mx.infer.rnn.one

Description

Inference for one-to-one fusedRNN (CUDA) models

Usage

mx.infer.rnn.one(

  infer.data,

  symbol,

  arg.params,

  aux.params,

  input.params = NULL,

  ctx = mx.cpu()

)

Arguments

Argument	Description
infer.data	Data iterator created by mx.io.bucket.iter
symbol	Symbol used for inference
ctx