i m a newbie on machine learning and doing a project.
But i m stuck in low computation speed on my code.
Above one is my simple geodesic distance calculation module.
The inputs are in a form with [batch,channel,vector] .
In general, batch,channel and vector sizes are 64,128,64 respectively.
Btw the calculation is very very slow.
i think it would be my fault, but i cannot find what is main reason of this phenomenon.
why is this module slow??
please help this novice.
following code is example of using above module.