einsum

차이

문서의 선택한 두 판 사이의 차이를 보여줍니다.

차이 보기로 링크

--- einsum [2018/08/21 06:22] – 만듦 rex8312
+++ einsum [2024/03/23 02:38] (현재) – 바깥 편집 127.0.0.1
@@ 줄 9: / 줄 9: @@
 np.einsum('bi,bj,bk->bijk', a1, a2, a3).reshape(2, -1).shape
+rs = np.einsum('bi,bj,bk->bijk', a1, a2, a3).reshape(2, -1)
+actions = rs.reshape(2, -1).argmax(axis=1)
+a1_ = rs // 4 // 2 % 3
+a2_ = rs // 4 % 2
+a3_ = rs % 4
+actions = list()
+for n in [4, 2, 3]:
+  actions.append(rs % n)
+  rs = rs // n
+actions = reversed(actions)
 </code>
+  * https://optimized-einsum.readthedocs.io/en/stable/index.html
+{{tag>einsum numpy}}

einsum.1534832576.txt.gz · 마지막으로 수정됨: 2024/03/23 02:37 (바깥 편집)