1.tf.argmax()
a = tf.constant([[1,5,3],
[4,2,6]])
sess = tf.Session()
print sess.run(tf.argmax(a, 0))#[1 0 1]
print sess.run(tf.argmax(a, 1))#[1 2]
#argmax()函数返回最大值的索引,0返回每一列最大值的索引,1返回每一行最大值的索引
2.tf.random_normal | tf.truncated_normal | tf.random_uniform
tf.random_normal(shape,mean=0.0,stddev=1.0,dtype=tf.float32,seed=None,name=None)
tf.truncated_normal(shape, mean=0.0, stddev=1.0, dtype=tf.float32, seed=None, name=None)
tf.random_uniform(shape,minval=0,maxval=None,dtype=tf.float32,seed=None,name=None)
这几个都是用于生成随机数tensor的。尺寸是shape
random_normal: 正太分布随机数,均值mean,标准差stddev
truncated_normal:截断正态分布随机数,均值mean,标准差stddev,不过只保留[mean-2*stddev,mean+2*stddev]范围内的随机数
random_uniform:均匀分布随机数,范围为[minval,maxval]
import tensorflow as tf
with tf.Session() as sess:
print "tf.random_normal([3,3,3], mean = 0.0, stddev = 2.0)----------------"
print sess.run(tf.random_normal([3,3,3], mean = 0.0, stddev = 2.0))
print "tf.random_uniform([3,3,3], minval = 0.0, maxval = 10)---------------"
print sess.run(tf.random_uniform([3,3,3], minval = 0.0, maxval = 10))
print "tf.truncated([2,2], mean = 5.0, stddev = 1.0)-------------------"
print sess.run(tf.truncated_normal([2,2], mean = 5.0, stddev = 1.0))
tf.random_normal([3,3,3], mean = 0.0, stddev = 2.0)----------------
[[[-1.52943552 -0.66951263 -0.92204839]
[ 0.98132163 0.30230746 -0.76994801]
[-2.52370524 1.63953972 -0.90987027]]
[[-2.49060225 2.71975255 1.97556829]
[ 1.31254244 -4.19041204 -2.62262273]
[ 1.17605031 -4.44652367 0.31839186]]
[[-1.66576338 1.9761579 -2.44654465]
[ 2.1690588 1.38606691 0.6156947 ]
[ 0.07768013 2.35041523 0.11098886]]]
tf.random_uniform([3,3,3], minval = 0.0, maxval = 10)---------------
[[[ 5.46428204 4.6617341 4.74864483]
[ 3.88854265 4.4059515 8.61946297]
[ 8.42585945 8.20820904 6.00255251]]
[[ 7.05990314 7.84427404 8.17808056]
[ 9.67488766 4.86453056 4.11894894]
[ 6.64579391 8.30910301 4.74615002]]
[[ 9.56317806 7.91035557 3.3411324 ]
[ 4.64870071 8.23911667 9.47863197]
[ 9.80347633 1.5449512 9.06284809]]]
tf.truncated([2,2], mean = 5.0, stddev = 1.0)-------------------
[[ 5.74013472 5.93944788]
[ 5.29522514 4.77231503]]