gotch/example/tensor-io/main.go

75 lines
1.2 KiB
Go

package main
import (
"fmt"
"git.andr3h3nriqu3s.com/andr3/gotch"
"git.andr3h3nriqu3s.com/andr3/gotch/ts"
)
func main() {
x, err := ts.OfSlice([]float64{1.3, 29.7})
if err != nil {
panic(err)
}
path := "file.pt"
x.MustSave(path)
loadedTs := ts.MustLoad(path)
loadedTs.Print()
ts1 := ts.MustOfSlice([]float64{1.3, 29.7})
ts2 := ts.MustOfSlice([]float64{2.1, 31.2})
var namedTensors []ts.NamedTensor = []ts.NamedTensor{
{Name: "ts1", Tensor: ts1},
{Name: "ts2", Tensor: ts2},
}
pathMulti := "file_multi.pt"
// err = ts.SaveMulti(namedTensors, pathMulti)
// if err != nil {
// panic(err)
// }
err = ts.SaveMultiNew(namedTensors, pathMulti)
if err != nil {
panic(err)
}
var data []ts.NamedTensor
data = ts.MustLoadMulti(pathMulti)
for _, v := range data {
v.Tensor.Print()
}
device := gotch.NewCuda()
data = ts.MustLoadMultiWithDevice(pathMulti, device)
for _, v := range data {
v.Tensor.Print()
}
tsString := x.MustToString(80)
fmt.Printf("Tensor String: \n%v\n", tsString)
imagePath := "mnist-sample.png"
imageTs, err := ts.LoadHwc(imagePath)
if err != nil {
panic(err)
}
err = imageTs.Save("mnist-tensor-saved.png")
if err != nil {
panic(err)
}
}