mindflow.pde.FlowWithLoss

class mindflow.pde.FlowWithLoss(model, loss_fn='mse')

基于数据驱动的流体问题求解的基类。

参数：

• model (mindspore.nn.Cell) - 用于训练或测试的网络模型。

• loss_fn (Union[str, Cell]) - 损失函数。默认值： 'mse'。

异常：

• TypeError - 如果 mode 或 loss_fn 的类型不是mindspore.nn.Cell。

• NotImplementedError - 如果成员函数 get_loss 未定义。

支持平台：

Ascend GPU

get_loss(inputs, labels)

计算训练或测试模型的损失。

参数：

• inputs (Tensor) - 网络模型的输入数据。

• labels (Tensor) - 样本的真值。