Python mxnet.autograd.predict_mode用法及代码示例

用法: mxnet.autograd.predict_mode()

返回要在 ‘with’ 语句中使用的范围上下文，其中前向传递行为设置为推理模式，而不更改记录状态。

例子：

with autograd.record():
    y = model(x)
    with autograd.predict_mode():
        y = sampling(y)
    backward([y])

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.autograd.predict_mode。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。