从键盘输入一个m*n的二维张量【tensorflow】

代码：

# 从键盘输入一个二维张量
import tensorflow as tf

sess = tf.InteractiveSession()

m = int(input())  # 输入一个数m，需强制类型转换
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])  # 声明一个tf常量
init_op = tf.global_variables_initializer()  # 全局变量初始化操作
sess.run(init_op)
A = sess.run(A)
print(A)

sess.close()