1.引入tensorflow
import tensorflow as tf
2.创建矩阵
import tensorflow as tf
x = tf.constant([[1,9],[3,6]])
x
输出为:
<tf.Tensor: shape=(2, 2), dtype=int32, numpy=
array([[1, 9],
[3, 6]], dtype=int32)>
tf.Tensor表示一个张量,shape=(2,2)表示矩阵为2*2;dtype=int32表示数据类型为整数
3.tf和numpy交互
(1)
import tensorflow as tf
x = tf.constant([[1,9],[3,6]])
x
x.numpy()
输出为:
array([[1, 9],
leix [3, 6]], dtype=int32)
(2)
import tensorflow as tf
import numpy as np
a = np.ones([2,2])
print(a)
a1 = tf.multiply(a,2)
a1
输出为:
[[1. 1.]
[1. 1.]]
<tf.Tensor: shape=(2, 2), dtype=float64, numpy=
array([[2., 2.],
[2., 2.]])>
即将a中所有元素乘2
4.数类型转换
import tensorflow as tf
x = tf.constant([[1,9],[3,6]])
x = tf.cast(x, tf.float32)
x
输出为:
<tf.Tensor: shape=(2, 2), dtype=float32, numpy=
array([[1., 9.],
[3., 6.]], dtype=float32)>
转成了单精度浮点数