欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

tensorflow笔记:常用函数

程序员文章站 2022-05-30 12:30:03
...

常用函数

1、tf.constant

tf.constant(value,dtype=None,shape=None,name=’Const’,verify_shape=False)
作用:创建一个常量tensor
参数:
value: 一个dtype类型(如果指定了)的常量值或list。
  如果value是一个list,len(value)不能够超过shape参数指定的大小(如果已指定);
  要是len(list)小于shape指定的大小,多余的则由list的最后一个元素进行填充。
dtype: 返回tensor的类型
shape: 指定tensor的形状。
name: tensor的名字
verify_shape:是否允许对值的形状进行验证。

import tensorflow as tf

#build graph
a = tf.constant(2,shape=[2],name='a')
b = tf.constant(2,shape=[2,2])
c = tf.constant([1,2,3],shape=[6])
d = tf.constant([1,2,3],shape=[3,2])

#construct session
sess=tf.Session()

#run in session
print "a:",a
print "name of a:",a.name
print sess.run(a)

print "b:",b
print "type of b:",type(b)
print sess.run(b)

print "c:",c
print "type of c:",type(c)
print sess.run(c)

print "d:",d
print "type of d:",type(d)
print sess.run(d)