fix: output shape #1

Closed senysenyseny16 closed 1 year ago