preparing the field for tensor cores transposes

2019-07-17 13:20:33 -07:00
parent d2e116d057
commit bfa39b8992
9 changed files with 67 additions and 29 deletions
--- a/examples/python/tensorflow/run.py
+++ b/examples/python/tensorflow/run.py
@@ -23,7 +23,7 @@ def run_dot():
    result = sess.run([c], feed_dict = {a: ha,
                                        b: hb})[0]
    # Test
-    hresult = np.dot(ha.T, hb).T
+    hresult = np.dot(ha.T, hb.T).T
    dif = np.abs(result - hresult)
    print(hresult)
    print(result)