Struct DiceInput

pub struct DiceInput<B, const D: usize = 4>
where
    B: Backend,
{ /* private fields */ }

Input type for the DiceMetric.

Type Parameters

• B: Backend type.
• D: Number of dimensions. Should be more than, or equal to 3 (default 4).

Implementations

impl<B, const D: usize> DiceInput<B, D>

where B: Backend,

pub fn new( outputs: Tensor<B, D, Int>, targets: Tensor<B, D, Int>, ) -> DiceInput<B, D>

Creates a new DiceInput with the given outputs and targets.

Inputs are expected to have the dimensions [B, C, ...] where B is the batch size, C is the number of classes, and ... represents additional dimensions (e.g., height, width for images).

If C is more than 1, the first class (index 0) is considered the background. Additionally, one-hot encoding is the responsibility of the caller.

Arguments

• outputs: The model outputs as a tensor.
• targets: The ground truth targets as a tensor.

Returns

A new instance of DiceInput.

Panics

• If D is less than 3.
• If outputs and targets do not have the same dimensions.
• If outputs or targets do not have exactly D dimensions.
• If outputs and targets do not have the same shape.