昇腾社区首页
中文
注册

ConstantLayer *AddConstant(Dims dimensions, WeightsBuf weights) noexcept;

函数功能

添加常量层。

函数原型

ConstantLayer *AddConstant(Dims dimensions, WeightsBuf weights) noexcept;

约束说明

  • dimensions的shape维度之积需要与weights的count保持一致,且dimensions维度之积不可超过UINT64类型的最大值。
  • dimensions与weights不可为空。
  • dimensions维度必须在[1, 8]。

参数说明

参数名

输入/输出

说明

dimensions

输入

常量层的维度。

weights

输入

常量层的权重。

返回值说明

返回新的常量层,如果添加失败则返回nullptr或抛出异常。