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

标签 tensorflow eigen

我正在 tensorflow 中构建一个 C++ 操作。它需要将 uint8 值转换为 Eigen::half 吗?但是,我没有找到执行此操作的函数。但我确实注意到有一个 half_to_float()。 Eigen 库中有类似的函数 uint8_to_float() 吗?

最佳答案

您可以直接从 uint8_t 转换为 Eigen::half:

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

转换可能不是很有效,但应该可以:https://godbolt.org/z/Wc_j3X

如果您经常需要这个,您可以考虑实现一个小型查找表。

关于tensorflow - 如何将 uint8 转换为 Eigen :half?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57997369/

相关文章:

c++ - 使用 Eigen 就地进行 LDLT 分解

python - tensorflow 中的张量形状

python - 属性错误 : 'Tensor' object has no attribute '_keras_shape' when creating a model in keras

tensorflow - TensorFlow 2 中带有导数的损失函数

Eigen3 张量切片,无需复制数据

c++ - 将稀疏 mat-vec-mult 的结果存储到预分配的 vector 中

tensorflow - 使用附加数据重新训练现有的基本 BERT 模型

python - sigmoid激活函数可以用来解决Keras中的回归问题吗?

c++ - 在 Eigen 3.1.0-alpha1 中返回稀疏矩阵

c++ - 在我的 Cmake 项目中使用 Eigen Lib?