Nishant sethi

can you please provide some description for this.

@vinayak-mehta thanks, I'm picking this up.

@vinayak-mehta Please check my PR

I would like to contribute to this

It should output 528 if we are computing square on axis=-1, which is not the requirement of this notebook. So, I feel this is the only reason why the output...

i'll take this and i will work on flask

@rafaelhenrique can you briefly describe what exactly is your expectation