huggingface / huggingface/candle

Implement `AvgPool1d` and `AdaptiveAvgPool1d`

Open
#1,664 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Rust
Stars
21k
Forks
1.8k
Avg merge
16h 42m
Merged PRs (30d)
25

Description

pytorch versions:
1. https://pytorch.org/docs/stable/generated/torch.nn.AvgPool1d.html
2. https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool1d.html

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reading the linked PyTorch documentation for AvgPool1d and AdaptiveAvgPool1d, then locate Candle's existing pooling operations and their tests. Done means both operations are implemented with behavior matching the two referenced PyTorch APIs and covered by appropriate tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
pytorch, rust
Domain
machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.