其他类

核生成器

class MinkowskiEngine.KernelGenerator(kernel_size=-1, stride=1, dilation=1, is_transpose: bool = False, region_type: MinkowskiEngineBackend._C.RegionType = <RegionType.HYPER_CUBE: 0>, region_offsets: Optional[torch.Tensor] = None, expand_coordinates: bool = False, axis_types=None, dimension=-1)
__init__(kernel_size=-1, stride=1, dilation=1, is_transpose: bool = False, region_type: MinkowskiEngineBackend._C.RegionType = <RegionType.HYPER_CUBE: 0>, region_offsets: Optional[torch.Tensor] = None, expand_coordinates: bool = False, axis_types=None, dimension=-1)

region_type (RegionType, optional): 定义内核形状。请参考 MinkowskiEngine.Comon 获取更多详情。

region_offset (torch.IntTensor, optional): 当 region_typeRegionType.CUSTOM 时,卷积核使用提供的 region_offset 来定义偏移量。它应该是一个大小为 \(N \times D\) 的矩阵,其中 \(N\) 是偏移量的数量,\(D\) 是空间的维度。

axis_types (RegionType 列表, optional): 如果给定,它使用不同的方法为每个轴创建内核。例如,当它是 [RegionType.HYPER_CUBE, RegionType.HYPER_CUBE, RegionType.HYPER_CROSS] 时,内核对于前两个维度将是矩形的,对于第三个维度将是十字形的。

axis_types
cache
dimension
expand_coordinates
get_kernel(tensor_stride, is_transpose)
kernel_dilation
kernel_size
kernel_stride
kernel_volume
region_offsets
region_type
requires_strided_coordinates

RegionType

class MinkowskiEngine.RegionType

成员

HYPER_CUBE

HYPER_CROSS

CUSTOM

CUSTOM = <RegionType.CUSTOM: 2>
HYPER_CROSS = <RegionType.HYPER_CROSS: 1>
HYPER_CUBE = <RegionType.HYPER_CUBE: 0>
property name