python - 如何获取tensorflow张量大小(以字节为单位)？

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 中获取更多信息。 。玩得开心!