FluxML/Flux.jl

Overloading Flux.batch for custom eltypes

Chiusa

#1821 aperta il 3 gen 2022

 (13 commenti) (0 reazioni) (0 assegnatari)Julia (619 fork)batch import
enhancementgood first issuehelp wanted

Metriche repository

Star
 (4725 stelle)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

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?

Guida contributor