FluxML/Flux.jl

Overloading Flux.batch for custom eltypes

Open

#1 821 ouverte le 3 janv. 2022

Voir sur GitHub
 (13 commentaires) (0 réactions) (0 assignés)Julia (619 forks)batch import
enhancementgood first issuehelp wanted

Métriques du dépôt

Stars
 (4 725 stars)
Métriques de merge PR
 (Merge moyen 4h 27m) (2 PRs mergées en 30 j)

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?

Guide contributeur