FluxML/Flux.jl

Overloading Flux.batch for custom eltypes

Open

#1.821 aberto em 3 de jan. de 2022

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

Métricas do repositório

Stars
 (4.725 stars)
Métricas de merge de PR
 (Mesclagem média 4h 27m) (2 fundiu PRs em 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?

Guia do colaborador