矩阵加法测试

# 矩阵加法测试
import tensorflow as tf

m = int(input())
n = int(input())
count = m * n
A_list = []
while count > 0:
    x_input = int(input())
    A_list.append(x_input)
    count = count - 1
A = tf.constant(A_list, shape=[m, n])
m = int(input())
n = int(input())
count = m * n
A_list.clear()
while count > 0:
    x_input = int(input())
    A_list.append(x_input)
    count = count - 1
B = tf.constant(A_list, shape=[m, n])
C = tf.add(A, B)
init_op = tf.global_variables_initializer()
sess = tf.InteractiveSession()
sess.run(init_op)
A = sess.run(A)
B = sess.run(B)
C = sess.run(C)
print(A)
print(B)
print(C)
sess.close()

猜你喜欢

转载自blog.csdn.net/xky1306102chenhong/article/details/81109543