接口名 |
概述 |
支持平台 |
sciai.architecture.AdaptActivation
|
具有可训练参数和固定尺度的自适应激活函数。 |
GPU CPU Ascend
|
sciai.architecture.FirstOutputCell
|
返回指定网络第一个输出的网络。 |
GPU CPU Ascend
|
sciai.architecture.FNO1D
|
一维傅里叶神经算子(FNO1D)包含一个提升层、多个傅里叶层和一个解码器层。 |
Ascend GPU
|
sciai.architecture.FNO2D
|
二维傅里叶神经算子(FNO2D)包含一个提升层、多个傅里叶层和一个解码器层。 |
Ascend GPU
|
sciai.architecture.FNO3D
|
三维傅里叶神经算子(FNO3D)包含一个提升层、多个傅里叶层和一个解码器层。 |
Ascend GPU
|
sciai.architecture.get_activation
|
根据名字获取对应激活函数。 |
Ascend GPU CPU
|
sciai.architecture.MLP
|
多层感知器。 |
Ascend GPU CPU
|
sciai.architecture.MLPAAF
|
带自适应激活函数的多层感知器。 |
GPU CPU Ascend
|
sciai.architecture.MLPShortcut
|
带有残差的多层感知器。 |
GPU CPU Ascend
|
sciai.architecture.MSE
|
与0的均方误差。 |
GPU CPU Ascend
|
sciai.architecture.NoArgNet
|
返回指定网络第一个输出的网络。 |
GPU CPU Ascend
|
sciai.architecture.Normalize
|
使用指定的下界和上界对输入进行归一化。 |
GPU CPU Ascend
|
sciai.architecture.PDENet
|
PDE-Net模型。 |
Ascend GPU
|
sciai.architecture.SReLU
|
Sin整流线性单元激活函数。 |
Ascend
|
sciai.architecture.SSE
|
与0的平方和误差。 |
GPU CPU Ascend
|
sciai.architecture.Swish
|
带反向传播的 Swish(Silu) 激活函数。 |
GPU CPU Ascend
|
sciai.architecture.ViT
|
该模块基于ViT,包括encoder层、decoding_embedding层、decoder层和dense层。 |
Ascend GPU
|