Scripts: Información sobre las entradas y salidas del modelo ONNX

[Automated-Trading]

Información sobre las entradas y salidas del modelo ONNX:

Script para obtener información sobre el número, los tipos y los tamaños de los tensores de entrada y salida del modelo onnx

Autor: Slava

[Jeferson Henrique Godoy Dos Santos]

Automated-Trading:
Script para obtener información sobre el número, tipos y tamaños de los tensores de entrada y salida del modelo onnx

Ejemplo de código Python que utiliza la biblioteca onnx para obtener información sobre los tensores de entrada y salida de un modelo ONNX:

Asegúrate de sustituir "ruta/de/modelo.onnx" por la ruta del archivo ONNX de tu modelo. Este código cargará el modelo ONNX, recorrerá los tensores de entrada y salida y mostrará sus nombres, tipos y formatos.

Espero que te haya servido de ayuda.

[floston]

¿falta el soporte?)

[Slava]

floston #:

¿falta el soporte?)

No. El formato ha cambiado. El script actualizado está siendo revisado por el moderador.

Aquí, prueba el script actualizado con el modelo de cluster iris (esa es la razón del cambio). Cambia el nombre a iris.onnx