如何在金属着色器中执行2个矢量的外产物？

Question

因此，我正在开发一个神经网络以在GPU上在iOS中运行的神经网络，因此，使用我需要的矩阵符号（为了将错误反向传播）能够执行2个向量的外产品。

// Outer product of vector A and Vector B
kernel void outerProduct(const device float *inVectorA [[ buffer(0) ]],
                         const device float *inVectorB [[ buffer(1) ]],
                         device float *outVector [[ buffer(2) ]],
                         uint id [[ thread_position_in_grid ]]) {
    
    outVector[id] = inVectorA[id] * inVectorB[***?***]; // How to find this position on the thread group (or grid)?
}

Answer 1

您正在使用thread_position_in_grid错误地。如果要派遣2D网格，则应是uint2或ushort2，否则仅获得x坐标。请参阅金属阴影语言规范）的表5.7。

我不确定我们在谈论哪种外部产品，但我认为输出应该是矩阵。如果您要线性存储它，则计算Outvector的代码应该看起来像这样：

kernel void outerProduct(const device float *inVectorA [[ buffer(0) ]],
                         const device float *inVectorB [[ buffer(1) ]],
                         uint2 gridSize [[ threads_per_grid ]],
                         device float *outVector [[ buffer(2) ]],
                         uint2 id [[ thread_position_in_grid ]]) {
    
    outVector[id.y * gridSize.x + id.x] = inVectorA[id.x] * inVectorB[id.y];
}

另外，如果您要派发网格的大小，则invectora xInvectorB ，您可以在内核参数上使用属性threads_per_grid以找出网格的大小。

另外，您可以将向量的大小与向量本身一起传递。

Answer 2

我惊讶地发现金属没有2D跨产品（又称内部产品），所以在这里

float cross( float2 A, float2 B )
{
    float2 C = A.xy * B.yx;  // <- note B's swizzle
    return C.x - C.y;
}

回答您的问题：

float X = cross( inVectorA.read( id ), inVectorB.read( id ) );
outVector.write( X, id );