diff --git a/tensor/print.go b/tensor/print.go index 61bb09b..d6a3796 100644 --- a/tensor/print.go +++ b/tensor/print.go @@ -43,7 +43,12 @@ func (ts *Tensor) ValueGo() interface{} { if err != nil { log.Fatal(err) } - // fmt.Println(dst) + + // convert []int32 -> int + if reflect.TypeOf(dst).String() == "[]int32" { + dst = sliceInt32ToInt(dst.([]int32)) + } + return dst } func (ts *Tensor) ToSlice() reflect.Value { diff --git a/tensor/util.go b/tensor/util.go index da7d601..cba7a79 100644 --- a/tensor/util.go +++ b/tensor/util.go @@ -490,3 +490,11 @@ func sliceIntToInt32(input []int) []int32 { } return out } + +func sliceInt32ToInt(input []int32) []int { + out := make([]int, len(input)) + for i := 0; i < len(input); i++ { + out[i] = int(input[i]) + } + return out +}