Struct PointTransformerConfig 

pub struct PointTransformerConfig {
    pub in_dim: usize,
    pub dim: usize,
    pub out_dim: usize,
    pub pos_hidden: usize,
    pub attn_hidden: usize,
    pub k: usize,
}

Configuration for a PointTransformerLayer.

Fields

in_dim: usize

Input feature dimension d_in.

dim: usize

Attention (query / key / value) dimension d.

out_dim: usize

Output feature dimension d_out.

pos_hidden: usize

Hidden width of the position-encoding MLP θ.

attn_hidden: usize

Hidden width of the attention MLP γ.

k: usize

Number of neighbours k (includes the point itself).

Implementations

impl PointTransformerConfig

pub fn new( in_dim: usize, dim: usize, out_dim: usize, pos_hidden: usize, attn_hidden: usize, k: usize, ) -> VisionResult<Self>

Create and validate a configuration.

Errors

• VisionError::InvalidEmbedDim if in_dim, dim or out_dim is 0.
• VisionError::EmptyInput if k == 0, pos_hidden == 0 or attn_hidden == 0.

pub fn tiny() -> Self

A tiny configuration for unit tests: in_dim=8, dim=8, out_dim=8, pos_hidden=8, attn_hidden=8, k=4.

Trait Implementations

impl Clone for PointTransformerConfig

fn clone(&self) -> PointTransformerConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PointTransformerConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for PointTransformerConfig

fn eq(&self, other: &PointTransformerConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for PointTransformerConfig