Question: unsqueeze: axes in not an []int64

[loeffel-io]

Hello @owulveryck,

i try to run this onnx model:

with this code:

package main

import (
	"fmt"
	"github.com/owulveryck/onnx-go"
	"github.com/owulveryck/onnx-go/backend/x/gorgonnx"
	"gorgonia.org/tensor"
	"io/ioutil"
	"log"
)

func main() {
	backend := gorgonnx.NewGraph()
	model := onnx.NewModel(backend)

	var err error
	var b []byte
	var output []tensor.Tensor

	if b, err = ioutil.ReadFile("model.onnx"); err != nil {
		log.Fatal(err)
	}

	if err = model.UnmarshalBinary(b); err != nil {
		log.Fatal(err)
	}

	var acosGroupTensor tensor.Tensor
	if acosGroupTensor, err = tensor.Argmax(
		tensor.New(tensor.WithShape(1, 5), tensor.Of(tensor.Float32), tensor.WithBacking([]float32{0, 1, 0, 0, 0})),
		1,
	); err != nil {
		log.Fatal(err)
	}

	if err = model.SetInput(0, acosGroupTensor); err != nil {
		log.Fatal(err)
	}

	var acosRatioTensor = tensor.New(tensor.WithShape(1, 4), tensor.Of(tensor.Float32), tensor.WithBacking([]float32{0, 0, -1, 0}))

	if err = model.SetInput(1, acosRatioTensor); err != nil {
		log.Fatal(err)
	}

	var salesRatioTensor = tensor.New(tensor.WithShape(1, 4), tensor.Of(tensor.Float32), tensor.WithBacking([]float32{0, 0, -1, 0}))

	if err = model.SetInput(2, salesRatioTensor); err != nil {
		log.Fatal(err)
	}

	if err = backend.Run(); err != nil {
		log.Fatal(err)
	}

	if output, err = model.GetOutputTensors(); err != nil {
		log.Fatal(err)
	}
	// write the first output to stdout
	fmt.Println(output[0])
}

It looks like i miss something pretty small - would love to get some help ❤️ thank you

[loeffel-io]

ref: https://github.com/owulveryck/onnx-go/blob/a1116f3e59d41e088d88670cd1e0683709882806/backend/x/gorgonnx/unsqueeze.go#L57

[owulveryck]

My guess is that the axes is a int64 and not a []int64 which causes the error.

I don't know why, but you can try to change the code in unsqueeze.go like this:

	a, ok := o.Attributes["axes"].(int64)
	if !ok {
		return errors.New("unsqueeze: axes in not an []int64")
	}
        a.Axes = []int64{a}

[loeffel-io]

Thanks @owulveryck, i created a workaround like this too but then i got stuck because the Cast operator is unimplemented.

I moved to https://github.com/triton-inference-server/server

I would leave this open to inform that the Unsqueeze operator seams unstable - feel free to close if not