matlab中的GPU计算

Question

我的matlab版本是2014a。 Norm 函数内置于 matlab 中的 gpuArray 函数中。当输入是 gpuArray 时，我认为它应该 return gpuArray class。但在我的 matlab 中，它 return 是两倍。谁能告诉我发生了什么事？

示例，

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

b 的类型是 "double" 而不是 "gpuArray"。

谢谢

Answer 1

不幸的是，如您所见，这是 gpuArray 实施中的一个不一致之处。一般来说，return 数据与输入数组的值相关的方法，例如 min、max、sum 等 return 值如 gpuArray，但 norm 没有。