使用TensorFlow执行一系列具有不同形状的矩阵乘法

时间:2019-05-10 07:32:18

标签: python tensorflow matrix

我遇到以下问题:

假设我们得到了矩阵矩阵乘法的列表,矩阵的形状如下:

[(3x2)*(2x3),
(5x2)*(2x5),
(4x2)*(2x4),
(2x2)*(2x2)]

如何使用GPU通过tf执行此矩阵矩阵乘法列表?我们如何堆叠不规则形状的数组(因为据我所知,这些形状必须相等才能稍后在例如tf.map_fn上应用?)?

在tf中使用GPU执行许多较小的(更)尺寸的矩阵矩阵乘法实际上有意义吗?通常,分配诸如矩阵乘法或特征分解的任务是一种很好的策略/启发式方法。在GPU上?

最诚挚的问候

0 个答案:

没有答案