如何以字节为单位获取张量流张量大小？

Question

假设我有一个非常简单的变量：

my_var = tf.get_variable("my_var", (100,))

我希望能够以字节为单位计算张量的大小（不是形状）。当然这取决于数据类型。

我在官方文档中没有找到任何方法。

Answer 1

您可以使用 slim 计算张量的字节大小。

以下代码：

import tensorflow as tf
from tensorflow.contrib import slim

my_var = tf.get_variable("my_var", (100,))
slim.model_analyzer.analyze_vars([my_var], print_info=True)

输出：

---------
Variables: name (type shape) [size]
---------
my_var:0 (float32_ref 100) [100, bytes: 400]
Total size of variables: 100
Total bytes of variables: 400

您当然可以在 slim github 中获得更多信息。玩得开心！

如何以字节为单位获取张量流张量大小？

How to get tensorflow Tensor size in bytes?

python

tensorflow

tensor