matlab中的GPU计算

Question

我的 matlab 版本是 2014a。范数函数内置于 matlab 中的 gpuArray 函数中。我认为当输入是 gpuArray 时它应该返回 gpuArray 类。但在我的matlab中，它返回双倍。谁能告诉我发生了什么？

例子，

a=gpuArray.randn(3,4);
b=norm(a)

b 的类型是“double”而不是“gpuArray”。

谢谢

Answer 1

不幸的是，正如您所观察到的，这是gpuArray 的实现不一致。一般而言，min、max、sum 等返回与输入数组的值相关的数据的方法返回值为gpuArray，但norm 不会。