tensorflow - 如何将 uint8 转换为 Eigen:half？

Question

我正在张量流中构建一个 c++ 操作。它需要将uint8值转换为Eigen::half？但是，我没有找到执行此操作的功能。但我确实注意到有一个half_to_float(). uint8_to_float()Eigen 库中 是否有类似的功能？

标签： tensorfloweigen

Answer 1

您可以直接从uint8_tto 转换Eigen::half：

Eigen::half foo(uint8_t x)
{
    return Eigen::half(x);
}

转换可能不是很有效，但它应该可以工作：https ://godbolt.org/z/Wc_j3X

如果你需要这个很多，你可以考虑实现一个小的查找表。