mindspore.nn.probability.bnn_layers.ConvReparam =============================================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/probability/mindspore.nn.probability.bnn_layers.ConvReparam.rst :alt: 查看源文件 .. py:class:: mindspore.nn.probability.bnn_layers.ConvReparam(in_channels, out_channels, kernel_size, stride=1, pad_mode="same", padding=0, dilation=1, group=1, has_bias=False, weight_prior_fn=NormalPrior, weight_posterior_fn=normal_post_fn, bias_prior_fn=NormalPrior, bias_posterior_fn=normal_post_fn) 具有重构化参数的卷积变分层。 更多相关信息可以查阅 `自动编码变分贝叶斯 `_ 相关论文。 参数: - **in_channels** (int) - 输入的channels :math:`C_{in}`。 - **out_channels** (int) - 输出的channels :math:`C_{out}`。 - **kernel_size** (Union[int, tuple[int]]) - 数据类型是 int 或2个 int 的元组。内核大小指定二维卷积窗口的高度和宽度。若为一个整数则高度和宽度均为该值,若为元组则两个值分别为高度和宽度。 - **stride** (Union[int, tuple[int]]) - 内核移动的距离,若是一个整数表示,则移动的高度和宽度都是步幅,或者两个整数的元组分别表示移动的高度和宽度。默认值: ``1`` 。 - **pad_mode** (str) - 指定填充模式。可选值是"same"、"valid"和"pad"。默认值: ``"same"`` 。 - same:采用补全方式。输出高度和宽度将与输入相同。将在水平和垂直方向上计算填充的总数,并尽可能均匀地分布在顶部和底部、左侧和右侧。否则,最后的额外填充将从底部和右侧完成。如果设置了此模式,则 `padding` 必须为0。 - valid:采用丢弃的方式。输出的可能最大高度和宽度将不带 padding 返回。多余的像素将被丢弃。如果设置了此模式,则 `padding` 必须为0。 - pad:输入两侧的隐式 padding。 `padding` 的值将被填充到输入 Tensor 边界。 `padding` 必须大于或等于0。 - **padding** (Union[int, tuple[int]]) - 输入两侧的隐式 padding 。默认值: ``0`` 。 - **dilation** (Union[int, tuple[int]]) - 数据类型是 int 或2个 int 的元组。该参数指定空洞卷积的空洞率。如果设置为k>1,将有k−1每个采样位置跳过的像素。它的值必须大于或等于1,并受输入的高度和宽度限制。默认值: ``1`` 。 - **group** (int) - 将过滤器拆分为组,`in_channels` 和 `out_channels` 必须能被组数整除。默认值: ``1`` 。 - **has_bias** (bool) - 指定层是否使用偏置向量。默认值: ``false`` 。 - **weight_prior_fn** (Cell) - 权重的先验分布。它必须返回一个 MindSpore 分布实例。默认值: ``NormalPrior`` 。(创建标准正态分布的一个实例)。当前版本仅支持正态分布。 - **weight_posterior_fn** (function) - 采样权重的后验分布。它必须是一个函数句柄,它返回一个 MindSpore 分布实例。默认值: ``normal_post_fn`` 。当前版本仅支持正态分布。 - **bias_prior_fn** (Cell) - 偏置向量的先验分布。它必须返回一个 MindSpore 分布实例。默认值: ``NormalPrior`` (创建标准正态分布的实例)。当前版本仅支持正态分布。 - **bias_posterior_fn** (function) - 采样偏差向量的后验分布。它必须是一个函数句柄,它返回一个 MindSpore 分布实例。默认值: ``normal_post_fn`` 。当前版本仅支持正态分布。 输入: - **input** (Tensor) - 输入 Tensor 的 shape 为 :math:`(N, C_{in})`。 输出: Tensor,输出张量的形状是 :math:`(N, C_{out}, H_{out}, W_{out})`。