h*n*m与h*m*n做矩阵乘法运算得到h*n*n的张量,复杂度如何计算
时间: 2023-09-24 08:07:03 浏览: 51
Alphatenso + 矩阵乘法发现算法
假设$h$为第一维的大小,$n$为第二维的大小,$m$为第三维的大小。对于h*n*m与h*m*n做矩阵乘法得到h*n*n的张量,其时间复杂度可以按照以下步骤计算:
1. 首先对两个矩阵做乘法操作,时间复杂度为$O(h*n*m)$。
2. 对乘积矩阵进行重塑操作,即将其变为h*n*n的张量,时间复杂度为$O(h*n*n)$。
因此,总的时间复杂度为$O(h*n*m + h*n*n)$。需要注意的是,在实际应用中,不同的实现方式可能会对时间复杂度造成影响。
阅读全文