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 |
---|---|
|
Data iterator created by mx.io.bucket.iter |
|
Symbol used for inference |
|