在 C++ 中将 ArrayFire 数组数据从 f64 转换为 f32

Question

我有一个 dtype=f64 的 ArrayFire 数组，我需要将其转换为 dtype=f32。除了转换为主机数组并通过执行static_cast 进行迭代之外，我不确定如何执行此操作。我确信有更好的方法。我找到了af_cast，但它似乎只是C接口的一部分。

// Example, double precision array
af::array A_double(100, 1, f64); // f64 = double precision

// Need to cast this to an f32 array somehow...
af::array A_single = cast<f32>(A_double); // <-- something like this?

Answer 1

答案竟然简单得令人尴尬。 ArrayFire 库有一个名为 as 的函数，我不知道它在文档中被忽略了。

// Example, double precision array
af::array A_double(100, 1, f64); // f64 = double precision

// Cast to an f32 array
af::array A_single = A_double.as(f32);

这是documentation的链接。