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)