search

prisma-ai / MPSX

GPU tensor framework with support for running ONNX models
MIT License
58 stars 6 forks source link

Fix reshape allowzero default logic #6

Closed shvul closed 1 year ago

shvul commented 1 year ago

Current reshaping function doesn't work well with zero axes parameters, handling here allowzero = false case as it's a default one