import tensorflow as tf
a = tf.constant([1, 5], dtype=tf.int64)
print( a)
tf.Tensor([1 5], shape=(2,), dtype=int64)
a为一个维度,shape=(2,)存在两个元素
tf.constant([1, 2, 3, 4, 5, 6], shape=[3, 2])
<tf.Tensor: shape=(3, 2), dtype=int32, numpy=
array([[1, 2],
[3, 4],
[5, 6]])>
shape为3行2列,
x1=tf.constant([1.,2.,3.],dtype=tf.float64)
print(x1)
x2=tf.cast(x1,tf.int32)
print(x2)
print("minimum of x2:", tf.reduce_min(x2))
tf.constant()定义常量。cast()数据类型转换
tf.Tensor([1. 2. 3.], shape=(3,), dtype=float64)
tf.Tensor([1 2 3], shape=(3,), dtype=int32)
minimum of x2: tf.Tensor(1, shape=(), dtype=int32)
a = np.arange(0, 5)
b = tf.convert_to_tensor(a, dtype=tf.int64)
print("a:", a)
print("b:", b)
a: [0 1 2 3 4]
b: tf.Tensor([0 1 2 3 4], shape=(5,), dtype=int64)
tf.convert_to_tensor(),标量转换为张量