mindflow.geometry.Cuboid
- class mindflow.geometry.Cuboid(name, coord_min, coord_max, dtype=numpy.float32, sampling_config=None)[源代码]
立方体对象的定义。
- 参数:
name (str) - 立方体对象的名称。
coord_min (Union[tuple[int, int], tuple[float, float], list[int, int], list[float, float], numpy.ndarray]) - 立方体对象左下角的坐标。
coord_max (Union[tuple[int, int], tuple[float, float], list[int, int], list[float, float], numpy.ndarray]) - 立方体对象右上角的坐标。
dtype (numpy.dtype) - 采样点数据类型的数据类型。默认值:
numpy.float32
。sampling_config (SamplingConfig) - 采样配置。默认值:
None
。
- 支持平台:
Ascend
GPU
样例:
>>> from mindflow.geometry import generate_sampling_config, Cuboid >>> cuboid_mesh = dict({'domain': dict({'random_sampling': False, 'size': [50, 50, 25]}), ... 'BC': dict({'random_sampling': False, 'size': 1000, 'with_normal': True,}),}) >>> cuboid = Cuboid("cuboid", (-3.0, 1, 0), (1, 2, 1), sampling_config=generate_sampling_config(cuboid_mesh)) >>> domain = cuboid.sampling(geom_type="domain") >>> bc, bc_normal = cuboid.sampling(geom_type="BC") >>> print(domain.shape) (62500, 3)