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 09e8768 commit 970599fCopy full SHA for 970599f
src/batchs.jl
@@ -35,7 +35,7 @@ end
35
get_slice(lengths::Vector{Int}, i::Integer) = (lengths[i]+1):lengths[i+1]
36
get_slice(lengths::Vector{Int}, i::UnitRange) = (lengths[minimum(i)]+1):lengths[maximum(i)+1]
37
function get_element((; field, lengths)::ConcatenatedBatch, i::Integer)
38
- view(field, fill(:, ndims(field) - 1)..., get_slice(lengths, i))
+ field[fill(:, ndims(field) - 1)..., get_slice(lengths, i)]
39
end
40
function get_element((; field, lengths)::ConcatenatedBatch, i::UnitRange)
41
idx = lengths[minimum(i):(maximum(i)+1)]
0 commit comments