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 1D Fourier Neural Operator, which usually contains a Lifting Layer, a Fourier Block Layer and a Projection Layer.

Ascend GPU

mindflow.cell.FNO2D

The 2D Fourier Neural Operator, which usually contains a Lifting Layer, a Fourier Block Layer and a Projection Layer.

Ascend GPU

mindflow.cell.FNO3D

The 3D Fourier Neural Operator, which usually contains a Lifting Layer, a Fourier Block Layer and a Projection 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.PeRCNN

Physics-embedded Recurrent Convolutional Neural Network (PeRCNN) Cell.

Ascend GPU

mindflow.cell.ResBlock

The ResBlock of dense layer.

Ascend GPU

mindflow.cell.UNet2D

The 2-dimensional U-Net model.

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