mindflow.cfd.cal_con_var

查看源文件
mindflow.cfd.cal_con_var(pri_var, material)[源代码]

从原始量中计算守恒量。

参数:
  • pri_var (Tensor) - 原始量。

  • material (mindflow.cfd.Material) - 流体材料。

返回:

Tensor,shape和 pri_var 一致。

支持平台:

GPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindflow.utils import cal_con_var
>>> from mindflow import material
>>> config =  {'type': 'IdealGas', 'heat_ratio': 1.4, 'specific_heat_ratio': 1.4, 'specific_gas_constant': 1.0}
>>> m = material.select_material(config)
>>> x1 = Tensor(np.random.randn(5, 32, 32, 1))
>>> x2 = cal_con_var(x, m)
>>> x2.shape
(5, 32, 32, 1)