mirrors/ollama-for-amd
1
0
Fork 0
mirror of https://github.com/likelovewant/ollama-for-amd.git synced 2025-12-21 14:26:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

ml: add slice operation (#12870)

Browse Source 
* slice

* chunk, chunksections
This commit is contained in:
Michael Yang
2025-11-13 13:28:21 -08:00
committed by GitHub
parent 482bec824f
commit b48083f33f
3 changed files with 773 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ml/backend.go
View File

@@ -198,6 +198,10 @@ type Tensor interface {
Copy(ctx Context, t2 Tensor) Tensor
Duplicate(ctx Context) Tensor
Slice(ctx Context, dim, low, high, step int) Tensor
Chunk(ctx Context, dim int, size int) []Tensor
ChunkSections(ctx Context, dim int, sections ...int) []Tensor
TopK(ctx Context, k int) Tensor
Argsort(ctx Context) Tensor
Mean(ctx Context) Tensor