昇腾社区首页
中文
注册
开发者
下载

InitializerType

功能描述

权重初始化类型枚举,定义嵌入表权重的初始化方式。

函数原型

1
2
3
4
class InitializerType(Enum):
     LINEAR ="linear"
     TRUNCATED_NORMAL ="truncated_normal"
     UNIFORM = "uniform"

参数说明

参数名

说明

LINEAR

线性初始化。

TRUNCATED_NORMAL

截断正态分布初始化。

UNIFORM

均匀分布初始化。

返回值说明

  • 成功:返回InitializerType枚举值。
  • 失败:抛出异常。