问题描述
我想将Plots.jl的绘图功能与通过ArrayFire的load_image()函数加载的图像一起使用。
我所拥有的是：
AFArray: 1000×300×3 Array{Float32,3}
我想要的是：
300×1000 Array{RGB{Any},2} with eltype RGB
我在文档中找不到直接转换。有什么有效的方法可以做到这一点吗？
解决方法
我不特别了解ArrayFire数组，但通常可以使用reinterpret进行此类操作。如果您希望新阵列驻留在CPU上，则将其复制过来。
然后，理想情况下，您可以这样做
rgb = reinterpret(RGB{Float32},A)
不幸的是，MxNx3不是RGB阵列的最佳布局，因为您希望3值顺序放置。因此，您应该确保阵列具有3xMxN布局，或者可以执行permutedims(A,(3,1,2))。
最后，要获得矩阵，必须删除前导单例维，否则将得到1xMxN数组。
所以
rgb = dropdims(reinterpret(RGB{Float32},permutedims(A,2))); dims=1)
我假设您实际上是要用RGB{Float32}而不是RGB{Any}。
顺便说一句，我不确定如果要将最终阵列保留在GPU上，将如何工作。
编辑：您可能考虑使用reshape而不是dropdims，这在我的电脑上似乎更快。