We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa25783 commit 0834ce8Copy full SHA for 0834ce8
src/batchs.jl
@@ -23,7 +23,7 @@ function ConcatenatedBatch((; field)::Batch)
23
end
24
function stack_ConcatenatedBatch(x::AbstractVector{<:ConcatenatedBatch})
25
field = reduce(x) do a, b
26
- cat(a, b; dims=ndims(a))
+ cat(a.field, b.field; dims=ndims(a.field))
27
28
offsets = vcat([0], getfield.(x, :lengths) .|> last)::Vector{Int} |> cumsum |> DropLast(1)
29
lengths = vcat([0], reduce(vcat,
0 commit comments