Skip to content

Commit 0834ce8

Browse files
committed
fix: fix issue
1 parent aa25783 commit 0834ce8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/batchs.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ function ConcatenatedBatch((; field)::Batch)
2323
end
2424
function stack_ConcatenatedBatch(x::AbstractVector{<:ConcatenatedBatch})
2525
field = reduce(x) do a, b
26-
cat(a, b; dims=ndims(a))
26+
cat(a.field, b.field; dims=ndims(a.field))
2727
end
2828
offsets = vcat([0], getfield.(x, :lengths) .|> last)::Vector{Int} |> cumsum |> DropLast(1)
2929
lengths = vcat([0], reduce(vcat,

0 commit comments

Comments
 (0)