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

Question

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

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

我希望能够计算张量的大小（而不是形状），以字节为单位。当然，它取决于dtype。

我在官方文档中没有发现任何方法。

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