TensorFlow是否支持类型为1位的二进制/布尔张量？

Question

TensorFlow是否支持布尔张量，其中每个元素只占1位（而不是8位）？我可以定义大张量~100 MB吗？

我在哪里可以阅读它？

在互联网上找不到任何搜索内容，这就是我要问的原因。

Answer 1

Tensorflow DT_BOOL张量映射到底层C ++ bool类型，该类型不占用1位内存。

您可以在Variant dtype上实现自己的操作，但需要大量的C ++编码。