以十为单位的稀疏到密集的einsum

2024-04-23 07:24:32 发布

您现在位置:Python中文网/ 问答频道 /正文

当我有复杂的操作时,有没有办法在稀疏矩阵上有效地执行操作?(事实上,我甚至不知道我的例子怎么能简化成没有切片的东西)

在这个例子中,我尝试用一个复杂的核向量来表示一个批(b)中的多个输入,这个核向量对于批中的每个例子都是不同的(因为它是由权重乘以另一个输入构建的)。输入是密集的,内核是稀疏的。你知道吗

示例:

outputs = special_math_ops.einsum("bi,bnilo->bnlo", x, kernel)

Tags: 示例切片矩阵mathoutputs内核向量ops