FluxML/Flux.jl

Overloading Flux.batch for custom eltypes

Open

#1,821 opened on Jan 3, 2022

View on GitHub
 (13 comments) (0 reactions) (0 assignees)Julia (619 forks)batch import
enhancementgood first issuehelp wanted

Repository metrics

Stars
 (4,725 stars)
PR merge metrics
 (Avg merge 3d 6h) (6 merged PRs in 30d)

Description

I ran into a use case for the batch function to operate on generators. I noticed that the batch function is not all that compatible with generators since it doesn't utilize the collect function for them. I'm curious if this additional functionality for the batch function is something that Flux.jl is interested in?

Contributor guide