mindflow.cell

API Name

Description

Supported Platforms

mindflow.cell.FCSequential

A sequential container of the dense layers, dense layers are added to the container sequentially.

Ascend GPU

mindflow.cell.FNO1D

The 1-dimensional Fourier Neural Operator (FNO1D) contains a lifting layer, multiple Fourier layers and a decoder layer.

Ascend GPU

mindflow.cell.FNO2D

The 2-dimensional Fourier Neural Operator (FNO2D) contains a lifting layer, multiple Fourier layers and a decoder layer.

Ascend GPU

mindflow.cell.InputScale

Scale the input value to specified region based on \((x_i - input_center)*input_scale\)

Ascend GPU

mindflow.cell.LinearBlock

The LinearBlock.

Ascend GPU

mindflow.cell.MultiScaleFCSequential

The multi-scale fully conneted network.

Ascend GPU

mindflow.cell.PDENet

The PDE-Net model.

Ascend GPU

mindflow.cell.ResBlock

The ResBlock of dense layer.

Ascend GPU

mindflow.cell.ViT

This module based on ViT backbone which including encoder, decoding_embedding, decoder and dense layer.

Ascend GPU

mindflow.cell.get_activation

Gets the activation function.

Ascend GPU