forward

crowdkit.learning.crowd_layer.CrowdLayer.forward | Source code

forward(
    self,
    outputs: Tensor,
    workers: Tensor
)

Forward pass.

Parameters description

ParametersTypeDescription
outputsTensor

Tensor of shape (batch_size, input_dim)

workersTensor

Tensor of shape (batch_size,) containing the worker IDs.

  • Returns:

    Tensor of shape (batch_size, num_labels)

  • Return type:

    Tensor

Last updated: March 31, 2023

Crowd-Kit
Overview
Reference
Aggregation
Datasets
Learning
Metrics
Postprocessing