211 Star 897 Fork 660

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
sciai.architecture.FNO1D.rst 1.71 KB
一键复制 编辑 原始数据 按行查看 历史
MaZhiming 提交于 2023-09-10 18:18 . [SciAI] init sciai docs

sciai.architecture.FNO1D

.. py:class:: sciai.architecture.FNO1D(in_channels, out_channels, resolution, modes, channels=20, depths=4, mlp_ratio=4, dtype=ms.float32)

    一维傅里叶神经算子(FNO1D)包含一个提升层、多个傅里叶层和一个解码器层。

    有关更多详细信息,请参考论文 `Fourier Neural Operator for Parametric Partial Differential Equations <https://arxiv.org/pdf/2010.08895.pdf>`_ 。

    参数:
        - **in_channels** (int) - 输入中的通道数。
        - **out_channels** (int) - 输出中的通道数。
        - **resolution** (int) - 输入的分辨率。
        - **modes** (int) - 要保留的低频分量的数量。
        - **channels** (int) - 输入提升尺寸后的通道数。默认值: `20`。
        - **depths** (int) - FNO层的数量。默认值: `4`。
        - **mlp_ratio** (int) - 解码器层的通道数提升比率。默认值: `4`。
        - **dtype** (dtype.Number) - 密集的计算类型。默认值: `ms.float32`。支持以下数据类型: `ms.float16` 或 `ms.float32`。GPU后端建议使用float32,Ascend后端建议使用float16。

    输入:
        - **x** (Tensor) - shape为 :math:`(batch\_size, resolution, in\_channels)` 的Tensor。

    输出:
        Tensor,FNO网络的输出。

        - **output** (Tensor) - shape为 :math:`(batch\_size, resolution, out\_channels)` 的Tensor。

    异常:
        - **TypeError** - 如果 `in_channels` 不是int。
        - **TypeError** - 如果 `out_channels` 不是int。
        - **TypeError** - 如果 `resolution` 不是int。
        - **TypeError** - 如果 `modes` 不是int。
        - **ValueError** - 如果 `modes` 小于1。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
r0.5

搜索帮助

0d507c66 1850385 C8b1a773 1850385