python输入变量输出常量_TensorFlow基本的常量、变量和运算操作详解

简介

深度学习需要熟悉使用一个框架，本人选择了TensorFlow，一边学习一边做项目，下面简要介绍TensorFlow中的基本常量、变量和运算操作，参考斯坦福大学的cs20si和TensorFlow官网API。

常量

tf.constant()

tf.constant(value, dtype=None, shape=None, name='Const', verify_shape=False),value为值，dtype类型，shape为张量形状，name名称、verify_shape默认False，这些项可选。作用创建一个常量。

a = tf.constant(2, name="a") # print(a) = 2

b = tf.constant(2.0, dtype=tf.float32, shape=[2,2], name="b") # 2x2矩阵，值为2

c = tf.constant([[1, 2], [3, 4]], name="c") # 2x2矩阵,值1,2,3,4

tf.zeros()和tf.zeros_like()

tf.zeros(shape, dtype=tf.float32, name=None), shape为张量形状，dtype类型，name名称。创建一个值为0的常量。

a = tf.zeros(shape=[2, 3], dtype=tf.int32, name='a') # 2x3矩阵，值为0, a = [[0, 0, 0], [0, 0, 0]]

tf.zeros_like(input_tensor, dtype=None, name=None, optimize=True)，input_tensor为张量，dtype类型，name名称，optimize优化。根据输入张量创建一个值为0的张量，形状和输入张量相同。

input_tensor = tf.constant([[1,2], [3,4], [5,6])

a = tf.zeros_like(input_tensor) # a = [[0, 0], [0, 0], [0, 0]]

tf.ones()和tf.ones_like()

tf.ones(shape, dtype=tf.float32, name=None)，与tf.zeros()类似。

tf.ones_like(input_tensor, dty