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 9739597 commit 798a5feCopy full SHA for 798a5fe
src/batchs.jl
@@ -24,7 +24,7 @@ end
24
function stack_ConcatenatedBatch(x::AbstractVector{<:ConcatenatedBatch})
25
field = cat(getfield.(x, :field), dims=ndims(first(x).field))
26
offsets = vcat([0], getfield.(x, :lengths) .|> last)::Vector{Int} |> cumsum
27
- lengths = zip(getfield.(x, :lengths), offsets) |> Map((lengths, offset) -> lengths .+ offsets) |> vcat
+ lengths = zip(getfield.(x, :lengths), offsets) |> Map((lengths, offset) -> lengths .+ offsets) |> vcat |> collect
28
ConcatenatedBatch(field, lengths)
29
end
30
get_slice(lengths::Vector{Int}, i::Integer) = (lengths[i]+1):lengths[i+1]
0 commit comments