使用cupy.einsum

编程问答 2022-06-12

问题描述

我正在处理Cupy的问题。我目前正在使用Cupy，并且在非常令人满意的高速下效果很好。但是当我使用cupy.einsum()方法时遇到问题 我在Numpy中使用相同的语法，没有任何错误。但是当使用Cupy时，会给我一个错误。这是代码部分

import numpy as np
A = np.random.randn(2,3,10)
B = np.random.randn(3,4)
C = np.einsum('ijk,jl->ijl',A,B)

这很好用，我得到了我一直想要的结果。但是，当我用Cupy编写相同的代码

import cupy as cp
A = cp.random.randn(2,10)
B = cp.random.randn(3,4)
C = cp.einsum('ijk,B)

运行此命令时，将计算A和B。但这在计算C时给了我一个错误。这是错误：

回溯（最近通话最近一次）：

文件“”，第4行，在 C = cp.einsum（'ijk，jl-> ijl'，A，B）

文件 “ C：\ Users \ Okan \ anaconda3 \ lib \ site-packages \ cupy \ linalg \ einsum.py”， einsum中的389行 如果dtype为None，则result_dtype = cupy.result_type（* operands）

文件“ array_function internals>”，第6行，在result_type

TypeError：未在类型上找到'numpy.result_type'的实现 实现 array_function 的示例：[]

如果您对此问题有想法或解决方案，我将非常高兴。 谢谢。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

cupynumpypython