在Python中torch.matmul计算不正确

Question

我遇到了一个惊人的问题。对于神经网络，我将两个矩阵与func save(_ request: JsonRepresentable, requestId: Int) { let requestData = ... requestData.data = request.toJSON() requestData.requestId = requestId requestData.timestamp = Date() save() } 相乘。 matrixA的大小为8x648（8批），矩阵B的大小为648x600。尽管每批都是相同的（也就是说，矩阵A的每一行都是相同的），但输出矩阵的第七行和第八行与其他六行不同。尽管它们之间只有微小的差异，但这对我的网络结果产生了巨大的影响。现在我不知道这怎么可能。令人惊讶的是，如果我分别计算matrixA.matmul(MatrixB)和matrixA[0].matmul(matrixB)，就不会出现问题。 有人知道为什么吗？