transcranial / keras-js

Run Keras models in the browser, with GPU support using WebGL

Home Page:https://transcranial.github.io/keras-js

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Merge layer , mode 'dot' implementation

aferdoash opened this issue · comments

I think the implementation of 'dot' mode for Merge layer ignores the fact that the dot axis can be a tuple too instead of an integer. When a tuple dotAxes is sent as part of attr to Merge layer for dot mode in Merge.js
this.dotAxes = dotAxes <= 0 ? dotAxes : dotAxes - 1 sets this.dotAxes to NaN

Thanks for catching this. Should be fixed now.