72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package ts_test
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.andr3h3nriqu3s.com/andr3/gotch"
|
|
"git.andr3h3nriqu3s.com/andr3/gotch/ts"
|
|
)
|
|
|
|
func ExampleTensor_MustArange() {
|
|
tensor := ts.MustArange(ts.FloatScalar(12), gotch.Int64, gotch.CPU).MustView([]int64{3, 4}, true)
|
|
|
|
fmt.Printf("%v", tensor)
|
|
|
|
// output
|
|
// 0 1 2 3
|
|
// 4 5 6 7
|
|
// 8 9 10 11
|
|
}
|
|
|
|
func ExampleTensor_Matmul() {
|
|
// Basic tensor operations
|
|
ts1 := ts.MustArange(ts.IntScalar(6), gotch.Int64, gotch.CPU).MustView([]int64{2, 3}, true)
|
|
defer ts1.MustDrop()
|
|
ts2 := ts.MustOnes([]int64{3, 4}, gotch.Int64, gotch.CPU)
|
|
defer ts2.MustDrop()
|
|
|
|
mul := ts1.MustMatmul(ts2, false)
|
|
defer mul.MustDrop()
|
|
fmt.Println("ts1: ")
|
|
ts1.Print()
|
|
fmt.Println("ts2: ")
|
|
ts2.Print()
|
|
fmt.Println("mul tensor (ts1 x ts2): ")
|
|
mul.Print()
|
|
|
|
//ts1:
|
|
// 0 1 2
|
|
// 3 4 5
|
|
//[ CPULongType{2,3} ]
|
|
//ts2:
|
|
// 1 1 1 1
|
|
// 1 1 1 1
|
|
// 1 1 1 1
|
|
//[ CPULongType{3,4} ]
|
|
//mul tensor (ts1 x ts2):
|
|
// 3 3 3 3
|
|
// 12 12 12 12
|
|
//[ CPULongType{2,4} ]
|
|
|
|
}
|
|
|
|
func ExampleTensor_AddScalar_() {
|
|
// In-place operation
|
|
ts3 := ts.MustOnes([]int64{2, 3}, gotch.Float, gotch.CPU)
|
|
fmt.Println("Before:")
|
|
ts3.Print()
|
|
ts3.MustAddScalar_(ts.FloatScalar(2.0))
|
|
fmt.Printf("After (ts3 + 2.0): \n")
|
|
ts3.Print()
|
|
ts3.MustDrop()
|
|
|
|
//Before:
|
|
// 1 1 1
|
|
// 1 1 1
|
|
//[ CPUFloatType{2,3} ]
|
|
//After (ts3 + 2.0):
|
|
// 3 3 3
|
|
// 3 3 3
|
|
//[ CPUFloatType{2,3} ]
|
|
}
|