Scripts: Información sobre las entradas y salidas del modelo ONNX
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.
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


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