Python 如何获取以字节为单位的tensorflow张量大小？

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

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

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

---

我在官方文件中找不到任何方法

您可以使用来计算张量的大小（以字节为单位）

以下代码：

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

---

当然，你可以在网上获得更多信息。玩得开心

那正是我要找的东西。非常感谢TF2，slim是一个独立的软件包：pip安装Tf slim