昇腾社区首页
中文
注册

def register

函数功能

用来注册时间步对应的策略。

函数原型

def register(self, step_range, strategy):

参数说明

参数名

输入/输出

类型

说明

step_range

输入

range、list或int

对应strategy的时间步,表示一定范围的时间步或某个时间步采用的策略,可以不设置所有时间步,不设置则采用默认策略。

strategy

输入

static或dynamic

对应输入时间步的策略,时间步默认是用dynamic策略。