You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Aug 11, 2020. It is now read-only.
This code base has been donated to the Apache MXNet project per #373, and repo is deprecated. Future development and issue tracking should continue in Apache MXNet.
Sign up for freeto subscribe to this conversation on GitHub.
Already have an account?
Sign in.
Hi!
The method BLASEngine<cpu,float>::gemm in MSHADOW_STAND_ALONE branch
contains the error in the shape of dst tensor(dot_engine-inl.h).
It is written
Tensor<cpu, 2, float> lhs((float*)B, Shape2(transpose_left ? k : n, transpose_left ? n : k)); // NOLINT()
Tensor<cpu, 2, float> rhs((float)A, Shape2(transpose_right ? m : k, transpose_right ? k : m)); // NOLINT(*)
Tensor<cpu, 2, float> dst(C, Shape2(m, n)); <-- (wrong shape)
Must be
Tensor<cpu, 2, float> dst(C, Shape2(n, m));
This error occurs when executing operator 'dot' and 'batch_dot' in JavaScript
The text was updated successfully, but these errors were encountered: