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