mirror of
https://github.com/wassname/keras-js.git
synced 2026-09-09 11:25:25 +08:00
fix dot/cos axes in Merge (close #18)
This commit is contained in:
@@ -720,8 +720,9 @@
|
||||
}
|
||||
],
|
||||
"metadata": {
|
||||
"anaconda-cloud": {},
|
||||
"kernelspec": {
|
||||
"display_name": "Python 3",
|
||||
"display_name": "Python [default]",
|
||||
"language": "python",
|
||||
"name": "python3"
|
||||
},
|
||||
|
||||
@@ -67,7 +67,7 @@ export default class Merge extends Layer {
|
||||
if (this.dotAxes < 0) {
|
||||
this.dotAxes = [shapes[0].length + this.dotAxes, shapes[1].length + this.dotAxes]
|
||||
} else {
|
||||
this.dotAxes = [this.dotAxes - 1, this.dotAxes - 1]
|
||||
this.dotAxes = [this.dotAxes, this.dotAxes]
|
||||
}
|
||||
}
|
||||
if (shapes[0][this.dotAxes[0]] !== shapes[1][this.dotAxes[1]]) {
|
||||
|
||||
Reference in New Issue
Block a user