各位同学好,今天和大家分享一下TensorFlow2.0中的tensor变量的创建方法。内容有:
(1) 通过numpy和list创建tensor;(2) 创建全部为某个值的tensor;(3) 随机化创建:正态分布、均匀分布;(4) 重新洗牌;(5) tf.constant()
那我们开始吧。
1. 通过numpy和list创建tensor
方法: tf.convert_to_tensor(value, dtype=None)
np.ones([2,3]) 使用numpy创建2行3列全为1的数组,将该数组全部变成tensor类型,指定数据类型维tf.float32
import tensorflow as tf
import numpy as np
#(1)通过numpy和list创建tensor
# 将2行3列全为1的numpy类型数据转换成tensor类型,原数据类型为float32
tf.convert_to_tensor(np.ones([2,3]),dtype=tf.float32)
# 将2行3列全为0的numpy类型数据转换成tensor类型,类型为float64
tf.convert_to_tensor(np.zeros([2,3]))
创建一个一维列表[2,3],和一个二维列表[[2],[2.]],转换成tensor类型
注意:此处的convert_to_tensor([2,3]) 括号内代表data一维数组[2,3],一定要和下面第2、3节的方法区分开
# 将1维列表变成tensor类型,int32
tf.convert_to_tensor([2,3]) #得到1维长度为2的tensor
# 将2维列表变成tensor类型,float32
tf.convert_to_tensor([[2],[2.]])
2. 创建全为某种值的tensor
2.1 创建全为0的tensor
方法: tf.zeros([shape])
注意:这里的