search

mlr-org / mlr3torch

Deep learning framework for the mlr3 ecosystem based on torch
https://mlr3torch.mlr-org.com
Other
40 stars 7 forks source link

`PipeOpTorch`: Add adaptive average pooling from torch #253

Closed sebffischer closed 1 month ago

sebffischer commented 4 months ago

Implement: https://torch.mlverse.org/docs/reference/torch_adaptive_avg_pool1d.html?q=adaptive#null for 1d, 2d and 3d.

This is similar to: https://github.com/mlr-org/mlr3torch/blob/main/R/PipeOpTorchAvgPool.R