Utils
AutoUMAP
__init__
add_auto_umap_args
on_train_start
plot
on_validation_end
OfflineUMAP
__init__
plot
Checkpointer
__init__
add_checkpointer_args
initial_setup
save_args
save
on_train_start
on_validation_end
Gather layer
Weighted KNN Classifier
__init__
update
compute
LARS
Metrics
accuracy_at_k
weighted_mean
Momentum module
MomentumUpdater
initialize_momentum_params
Sinkhorn-Knopp
Whitening
PositionalEncoding1D
PositionalEncoding1D applies positional encoding to the last dimension of a 3D tensor.
__init__
forward
PositionalEncodingPermute1D
PositionalEncodingPermute1D permutes the input tensor and applies 1D positional encoding.
__init__
forward
PositionalEncoding2D
PositionalEncoding2D applies positional encoding to the last two dimensions of a 4D tensor.
__init__
forward
PositionalEncodingPermute2D
PositionalEncodingPermute2D permutes the input tensor and applies 2D positional encoding.
__init__
forward
PositionalEncoding3D
PositionalEncoding3D applies positional encoding to the last three dimensions of a 5D tensor.
__init__
forward
PositionalEncodingPermute3D
PositionalEncodingPermute3D permutes the input tensor and applies 3D positional encoding.
__init__
forward
Summer
Summer adds positional encoding to the original tensor.