Skip to content

GitLab

Commit e0cd5116 authored by Michael Yang's avatar Michael Yang Committed by Michael Yang
Browse files

fix test

parent 20733207
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
convert/tensor.go
View file @ e0cd5116
...@@ -35,7 +35,7 @@ func splitDim(t Tensor, dim int, splits ...split) iter.Seq[*ggml.Tensor] { ...@@ -35,7 +35,7 @@ func splitDim(t Tensor, dim int, splits ...split) iter.Seq[*ggml.Tensor] {
slice := split.slices slice := split.slices
if len(slice) == 0 { if len(slice) == 0 {
slice := slices.Repeat([]tensor.Slice{nil}, len(shape)) slice = slices.Repeat([]tensor.Slice{nil}, len(shape))
slice[dim] = tensor.S(offset, offset+int(shape[dim])) slice[dim] = tensor.S(offset, offset+int(shape[dim]))
offset += int(shape[dim]) offset += int(shape[dim])
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment