fixed tensor.ValueGo() return []int

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 {

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
+}