方法:
num_elements()
功能:返回一个变量的规模,也就是该变量的所有元素的数量。
这个变量应该是该类型:
tensorflow.python.framework.tensor_shape.TensorShape
也可以看成是tensorflow中tensor数据类型的get_shape()的返回类型
代码例子:
import tensorflow as tf
import numpy as np
weights = tf.Variable(tf.zeros([32, 10]), name='Weight')
print(weights.get_shape()[:2].num_elements())
print(type(W.get_shape()[:2]))
结果:可以看出一起有32*10个数据