Python 如何在GPU上乘以矩阵列表
我有4个2x2复杂矩阵的列表(A,B,C,D),形状(大小,2,2)。 我需要计算以下表达式F=A.B.C.D,其中F具有相同的形状(大小,2,2)。 我只需要在Numpy中实现@operator就可以了,但需要在CPU上实现。 GPU上是否有任何运算符或函数?请尝试。Python 如何在GPU上乘以矩阵列表,python,numpy,gpu,matrix-multiplication,Python,Numpy,Gpu,Matrix Multiplication,我有4个2x2复杂矩阵的列表(A,B,C,D),形状(大小,2,2)。 我需要计算以下表达式F=A.B.C.D,其中F具有相同的形状(大小,2,2)。 我只需要在Numpy中实现@operator就可以了,但需要在CPU上实现。 GPU上是否有任何运算符或函数?请尝试。