commit
c27cb154d5
|
@ -12,7 +12,7 @@ mindspore.nn.probability.bnn_layers.DenseLocalReparam
|
||||||
.. math::
|
.. math::
|
||||||
\text{outputs} = \text{activation}(\text{inputs} * \text{weight} + \text{bias}),
|
\text{outputs} = \text{activation}(\text{inputs} * \text{weight} + \text{bias}),
|
||||||
|
|
||||||
此公式中,activation 为激活函数(若`activation` 参数传入),是与创建层的输入具有相同数据类型的权重矩阵。weight 是从权重的后验分布采样的权重矩阵。bias 是与由层创建的输入具有相同数据类型的偏置向量(仅当 `has_bias` 为 True 时),从 bias 的后验分布中采样。
|
此公式中,activation 为激活函数(若 `activation` 参数传入),是与创建层的输入具有相同数据类型的权重矩阵。weight 是从权重的后验分布采样的权重矩阵。bias 是与由层创建的输入具有相同数据类型的偏置向量(仅当 `has_bias` 为 True 时),从 bias 的后验分布中采样。
|
||||||
|
|
||||||
**参数:**
|
**参数:**
|
||||||
|
|
||||||
|
|
|
@ -13,7 +13,7 @@ mindspore.nn.probability.bnn_layers.DenseReparam
|
||||||
|
|
||||||
\text{outputs} = \text{activation}(\text{inputs} * \text{weight} + \text{bias}),
|
\text{outputs} = \text{activation}(\text{inputs} * \text{weight} + \text{bias}),
|
||||||
|
|
||||||
此公式中,activation 为激活函数(若`activation` 参数传入),是与创建层的输入具有相同数据类型的权重矩阵。weight 是从权重的后验分布采样的权重矩阵。bias 是与由层创建的输入具有相同数据类型的偏置向量(仅当 `has_bias` 为 True 时),从 bias 的后验分布中采样。
|
此公式中,activation 为激活函数(若 `activation` 参数传入),是与创建层的输入具有相同数据类型的权重矩阵。weight 是从权重的后验分布采样的权重矩阵。bias 是与由层创建的输入具有相同数据类型的偏置向量(仅当 `has_bias` 为 True 时),从 bias 的后验分布中采样。
|
||||||
|
|
||||||
**参数:**
|
**参数:**
|
||||||
|
|
||||||
|
|
|
@ -1,13 +1,13 @@
|
||||||
mindspore.nn.probability.bnn_layers.NormalPrior
|
mindspore.nn.probability.bnn_layers.NormalPrior
|
||||||
===============================================
|
===============================================
|
||||||
|
|
||||||
.. py:class:: mindspore.nn.probability.bnn_layers.NormalPrior(dtype=mindspore.float32, mean=0, std=0.1)
|
.. py:class:: mindspore.nn.probability.bnn_layers.NormalPrior(dtype=mstype.float32, mean=0, std=0.1)
|
||||||
|
|
||||||
初始化均值 0 和标准差 0.1 的正态分布。
|
初始化均值 0 和标准差 0.1 的正态分布。
|
||||||
|
|
||||||
**参数:**
|
**参数:**
|
||||||
|
|
||||||
- **dtype** (:class:`mindspore.dtype`) – 用于定义输出 Tensor 的数据类型的参数。默认值:mindspore.float32。
|
- **dtype** (:class:`mindspore.dtype`) – 用于定义输出 Tensor 的数据类型的参数。默认值:mstype.float32。
|
||||||
- **mean** (int, float) - 正态分布的平均值。默认值:0。
|
- **mean** (int, float) - 正态分布的平均值。默认值:0。
|
||||||
- **std** (int, float) - 正态分布的标准差。默认值:0.1。
|
- **std** (int, float) - 正态分布的标准差。默认值:0.1。
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue