如何获取tf.Tensor的大小(以位为单位)?

时间:2018-07-18 15:38:14

标签: python tensorflow

我可以使用tf.shape()获得张量的形状,并且可以类似地使用dtype属性来查找数据类型。我可以将张量的大小沿每个维度乘以数据类型所需的位数(例如tf.float32使用32位)来找出张量在内存中占用的空间,但是有一种更简单的方法可以这样做吗?

0 个答案:

没有答案