Tensorflow还原错误: Assign要求两个张量的形状匹配

import tensorflow as tf

# 创建两个张量
tensor1 = tf.constant([[1, 2], [3, 4]])
tensor2 = tf.constant([5, 6])

# 尝试将tensor2的值赋给tensor1
try:
    assign_op = tf.assign(tensor1, tensor2)
    with tf.Session() as sess:
        sess.run(assign_op)
except tf.errors.InvalidArgumentError as e:
    print("Error: Assign requires both tensors to have the same shape.")
    print("Details:", str(e))
    # 可以在这里添加代码来调整张量的形状，使其匹配

# 调整tensor2的形状，使其与tensor1匹配
tensor2_reshaped = tf.reshape(tensor2, [2, 1])

# 再次尝试将tensor2_reshaped的值赋给tensor1
assign_op = tf.assign(tensor1, tensor2_reshaped)
with tf.Session() as sess:
    sess.run(assign_op)
    print("tensor1 after assignment:")
    print(sess.run(tensor1))