tensorflow创建tensor

tensorflow创建张量tensor

import tensorflow as tf #导入tensorflow

import numpy as np #导入numpy

arangeNum = np.arange(0, 5) #创建列表

tensor = tf.convert_to_tensor(arangeNum, dtype=tf.int64) #将列表转换为tensor类型

zerosTensor = tf.zeros([3, 3]) #创建3*3矩阵，结果全为0的tensor

onesTensor = tf.ones(3) #创建全为1的tensor

fillTensor = tf.fill([3, 3], 0) #创建全为0的3*3的二维矩阵tensor

print(arangeNum) #打印列表

print(tensor) #打印tensor

print(zerosTensor) #打印3*3矩阵，结果全为0的tensor

print(onesTensor) #打印创建全为1的tensor

print(fillTensor) #打印全为0的3*3的二维矩阵tensor

结果： 

[0 1 2 3 4]#打印列表

tf.Tensor([0 1 2 3 4], shape=(5,), dtype=int64)#打印tensor

tf.Tensor(
[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]], shape=(3, 3), dtype=float32) #打印3*3矩阵，结果全为0的tensor

tf.Tensor([1. 1. 1.], shape=(3,), dtype=float32) #打印创建全为1的tensor

tf.Tensor(
[[0 0 0]
 [0 0 0]
 [0 0 0]], shape=(3, 3), dtype=int32) #打印全为0的3*3的二维矩阵tensor