创建常量tensor
常量tensor的创建方式比较多,常见的有一下几种方式:
tf.constant():创建常量tensor;
tf.zeros(), tf.zeros_like(), tf.ones(),tf.ones_like(): 创建全零或者全一的常量tensor;
tf.fill(): 创建自定义数值的tensor;
tf.random: 创建已知分布的tensor;
从numpy,list对象创建,再利用tf.convert_to_tensor转换为类型。
步骤 1tf.constant()
tf.constant(value, dtype=None, shape=None, name='Const', verify_shape=False):
value:值;
dtype:数据类型;
shape:张量形状;
name:常量名称;
verify_shape:布尔值,用于验证值的形状,默认False。verify_shape为True的话表示检查value的形状与shape是否相符,如果不符会报错。
代码:
const_a = tf.constant([[1, 2, 3, 4]],shape=[2,2], dtype=tf.float32) # 创建2x2矩阵,值1,2,3,4
const_a
输出:
代码:
查看常见属性
print("常量const_a的数值为:", const_a.numpy())
print("常量const_a的数据类型为:", const_a.dtype)
print("常量const_a的形状为:", const_a.shape)
print("常量const_a将被产生的设备名称为:", const_a.device)
输出:
常量const_a的数值为: [[1. 2.]
[3. 4.]]
常量const_a的数据类型为:
常量const_a的形状为: (2, 2)
常量const_a将被产生的设备名称为: /job:localhost/replica:0/task:0/device:CPU:0