使用TENSORFLOW创建两个张量,进行加减乘除运算并输出结果
时间: 2024-09-23 12:07:24 浏览: 61
在TensorFlow中,你可以使用`tf.constant`函数来创建常量张量,然后通过基本的算术运算符来进行加、减、乘、除等操作。这里是一个简单的例子:
```python
import tensorflow as tf
# 创建两个张量
tensor1 = tf.constant([10, 20], dtype=tf.float32) # [10.0, 20.0]
tensor2 = tf.constant([5, 10], dtype=tf.float32) # [5.0, 10.0]
# 进行加法运算
addition = tf.add(tensor1, tensor2)
print("Addition result:", addition.numpy()) # 输出: [15.0, 30.0]
# 减法运算
subtraction = tf.subtract(tensor1, tensor2)
print("Subtraction result:", subtraction.numpy()) # 输出: [5.0, 10.0]
# 乘法运算
multiplication = tf.multiply(tensor1, tensor2)
print("Multiplication result:", multiplication.numpy()) # 输出: [50.0, 200.0]
# 除法运算,注意这里是浮点数除法,整数除会有截断
division = tf.divide(tensor1, tensor2)
print("Division (float) result:", division.numpy()) # 输出: [2.0, 2.0]
```
以上代码首先导入了TensorFlow库,接着创建了两个形状相同的浮点型张量,然后分别进行了加、减、乘、除四种基本数学运算,并打印出相应的结果。
阅读全文