子矩阵乘法

问题描述 投票:0回答:0

我需要做更多次子矩阵乘法(来自相同的两个矩阵),其中子矩阵的维度和偏移量在每次迭代时都会发生变化。你知道一个合适的 cublas 函数来解决它吗?

我看到了 gemm 函数,但它似乎只适用于固定尺寸。

cublas
© www.soinside.com 2019 - 2024. All rights reserved.