python+ tensorflow : num_elements()

版权声明:本文系作者原创,未经允许,禁止转载。 https://blog.csdn.net/qq_27261889/article/details/82839314

方法:
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个数据
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_27261889/article/details/82839314