221 Star 941 Fork 692

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindchemistry.e3.nn.FullyConnectedNet.rst 1.02 KB
一键复制 编辑 原始数据 按行查看 历史
CaoWenbin 提交于 2024-08-16 17:01 +08:00 . API修改

mindchemistry.e3.nn.FullyConnectedNet

.. py:class:: mindchemistry.e3.nn.FullyConnectedNet(h_list, act=None, out_act=False, init_method='normal', dtype=float32)

    带有标量归一化激活的全连接神经网络。

    参数:
        - **h_list** (List[int]) - 用于密集层的输入、内部和输出维度的列表。
        - **act** (Func) - 将自动归一化的激活函数。默认值:``None``。
        - **out_act** (bool) - 是否对输出应用激活函数。默认值:``False``。
        - **init_method** (Union[str, mindspore.common.initializer]) - 初始化参数的方法。默认值:``'normal'`` 。
        - **dtype** (mindspore.dtype) - 输入张量的类型。默认值:``mindspore.float32`` 。

    输入:
        - **input** (Tensor) - 形状为 :math:`(h\_list[0])` 的张量。

    输出:
        - **output** (Tensor) - 形状为 :math:`(h\_list[-1])` 的张量。

    异常:
        - **TypeError**: 如果 `h_list` 的元素不是 `int`。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
r0.7

搜索帮助