metal - 如何在 Metal Performance Shaders 中实现矩阵所有元素的标量运算？

Question

假设我有一个矩阵，我想计算所有元素的 sin()。

我如何使用金属性能着色器做到这一点？

我是否编写了一些继承自的新内核MPSMatrixUnaryKernel？

Answer 1

我认为 MPS 并不意味着要扩展。您可以“简单地”为此编写自己的 Metal 计算着色器。

这似乎是一个很好的入门教程。