You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.common.initializer.rst 6.1 kB

4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. mindspore.common.initializer
  2. =============================
  3. 初始化神经元参数。
  4. .. py:class:: mindspore.common.initializer.Initializer(**kwargs)
  5. 初始化器的抽象基类。
  6. **参数:**
  7. - **kwargs** (`dict`) – **Initializer** 的关键字参数。
  8. .. py:method:: mindspore.common.initializer.initializer(init, shape=None, dtype=mstype.float32)
  9. 创建并初始化一个Tensor。
  10. **参数:**
  11. - **init** (`Union[Tensor, str, Initializer, numbers.Number]`) – 初始化方式。
  12. - *str*:`init` 是继承自 Initializer 的类的别名,实际使用时会调用相应的类。`init` 的值可以是"normal"、"ones"或"zeros"等。
  13. - *Initializer*:`init` 是继承自Initializer,用于初始化Tensor的类。
  14. - *numbers.Number*:调用常量来初始化张量。
  15. - **shape** (`Union[[tuple, list, int]`) - 被初始化的Tensor的shape,默认值为None。
  16. - **dtype** (`mindspore.dtype`) – 被初始化的Tensor的数据类型,默认值为 `mindspore.float32` 。
  17. **返回:**
  18. Tensor,返回一个张量对象。
  19. **异常:**
  20. TypeError: 参数`init`的类型不正确。
  21. ValueError: 通过`init`传入的Tensor的shape和作为参数传入的shape不一致。
  22. **样例:**
  23. .. code-block::
  24. >>> import mindspore
  25. >>> from mindspore.common.initializer import initializer, One
  26. >>> tensor = initializer('ones', [1, 2, 3], mindspore.float32)
  27. >>> tensor = initializer(One(), [1, 2, 3], mindspore.float32)
  28. >>> tensor = initializer(0, [1, 2, 3], mindspore.float32)
  29. .. py:class:: mindspore.common.initializer.TruncatedNormal(sigma=0.01)
  30. 生成一个数组用于初始化Tensor,数组中的数值从截断正态分布中采样得到。
  31. **参数:**
  32. - **sigma** (`float`) - 截断正态分布的标准差,默认值为0.01。
  33. .. py:class:: mindspore.common.initializer.Normal(sigma=0.01, mean=0.0)
  34. 生成一个数组用于初始化Tensor,数组中的数值从正态分布N(sigma, mean)中采样得到。
  35. .. math::
  36. f(x) = \frac{1} {\sqrt{2*π} * sigma}exp(-\frac{(x - mean)^2} {2*{sigma}^2})
  37. **参数:**
  38. - **sigma** (`float`) - 正态分布的标准差,默认值为0.01。
  39. - **mean** (`float`) - 正态分布的均值,默认值为0.0。
  40. .. py:class:: mindspore.common.initializer.Uniform(scale=0.07)
  41. 生成一个数组用于初始化Tensor,数组中的数值从均匀分布U(-scale, scale)中采样得到。
  42. **参数:**
  43. - **scale** (`float`) - 均匀分布的边界,默认值为0.07。
  44. .. py:class:: mindspore.common.initializer.HeUniform(negative_slope=0, mode="fan_in", nonlinearity="leaky_relu")
  45. 生成一个数组用于初始化Tensor,数组中的数值从HeKaiming均匀分布U[-boundary,boundary]中采样得到,其中
  46. .. math::
  47. boundary = \sqrt{\frac{6}{(1 + a^2) \times \text{fan_in}}}
  48. 是HeUniform分布的边界。
  49. **参数:**
  50. - **negative_slope** (`int, float, bool`) - 本层激活函数的负数区间斜率(仅适用于非线性激活函数"leaky_relu"),默认值为0。
  51. - **mode** (`str`) - 可选"fan_in"或"fan_out","fan_in"会保留前向传递中权重方差的量级,"fan_out"会保留反向传递的量级,默认为"fan_in"。
  52. - **nonlinearity** (`str`) - 非线性激活函数,推荐使用"relu"或"leaky_relu",默认为"leaky_relu"。
  53. .. py:class:: mindspore.common.initializer.HeNormal(negative_slope=0, mode="fan_in", nonlinearity="leaky_relu")
  54. 生成一个数组用于初始化Tensor,数组中的数值从HeKaiming正态分布N(0, sigma^2)中采样得到,其中
  55. .. math::
  56. sigma = \frac{gain} {\sqrt{N}}
  57. 其中,gain是一个可选的缩放因子。如果mode是"fan_in", N是权重Tensor中输入单元的数量,如果mode是"fan_out", N是权重Tensor中输出单元的数量。
  58. HeUniform 算法的详细信息,请查看 https://arxiv.org/abs/1502.01852。
  59. **参数:**
  60. - **negative_slope** (`int, float, bool`) - 本层激活函数的负数区间斜率(仅适用于非线性激活函数"leaky_relu"),默认值为0。
  61. - **mode** (`str`) - 可选"fan_in"或"fan_out","fan_in"会保留前向传递中权重方差的量级,"fan_out"会保留反向传递的量级,默认为"fan_in"。
  62. - **nonlinearity** (`str`) - 非线性激活函数,推荐使用"relu"或"leaky_relu",默认为"leaky_relu"。
  63. .. py:class:: mindspore.common.initializer.XavierUniform(gain=1)
  64. 生成一个数组用于初始化Tensor,数组中的数值从Xarvier均匀分布U[-boundary,boundary]中采样得到,其中
  65. .. math::
  66. boundary = gain * \sqrt{\frac{6}{n_{in} + n_{out}}}
  67. - `gain` 是一个可选的缩放因子。
  68. - `n_{in}` 为权重Tensor中输入单元的数量。
  69. - `n_{out}` 为权重Tensor中输出单元的数量。
  70. 有关 XavierUniform 算法的详细信息,请查看 http://proceedings.mlr.press/v9/glorot10a.html。
  71. **参数:**
  72. - **gain** (`float`) - 可选的缩放因子,默认值为1。
  73. .. py:class:: mindspore.common.initializer.One(**kwargs)
  74. 生成一个值全为1的常量数组用于初始化Tensor。
  75. .. py:class:: mindspore.common.initializer.Zero(**kwargs)
  76. 生成一个值全为0的常量数组用于初始化Tensor。
  77. .. py:class:: mindspore.common.initializer.Constant(value)
  78. 生成一个常量数组用于初始化Tensor。
  79. **参数:**
  80. - **value** (`Union[int, numpy.ndarray]`) - 用于初始化的常数值或者数组。