mindflow.cell

接口名

概述

支持平台

mindflow.cell.FCSequential

一个全连接层的顺序容器,按序放入全连接层。

Ascend GPU

mindflow.cell.FNO1D

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

Ascend GPU

mindflow.cell.FNO2D

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

Ascend GPU

mindflow.cell.InputScale

将输入值缩放到指定的区域。

Ascend GPU

mindflow.cell.LinearBlock

全连接模块。

Ascend GPU

mindflow.cell.MultiScaleFCSequential

多尺度的全连接神经网络。

Ascend GPU

mindflow.cell.PDENet

PDE-Net模型。

Ascend GPU

mindflow.cell.ResBlock

密集层的残差模块。

Ascend GPU

mindflow.cell.ViT

该模块基于ViT,包括encoder层、decoding_embedding层、decoder层和dense层。

Ascend GPU

mindflow.cell.get_activation

获取激活函数。

Ascend GPU