Table Of Contents
Table Of Contents

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