matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-10 07:46:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

arange

Browse Source
This commit is contained in:
Michael Yang
2025-04-03 10:25:23 -07:00
committed by Michael Yang
parent 1d99451ad7
commit 40b8fdbdca
5 changed files with 42 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ml/backend.go
View File

@@ -95,6 +95,9 @@ type Context interface {
FromFloatSlice(s []float32, shape ...int) (Tensor, error)
FromIntSlice(s []int32, shape ...int) (Tensor, error)
// Arange creates a 1D tensor with values within an interval (start, stop] increased by step.
Arange(start, stop, step float32, dtype DType) Tensor
Forward(...Tensor) Context
Compute(...Tensor)