tensorflow中面片的点积

import tensorflow as tf

def compute_dot_product(face1, face2):
    """
    计算两个面片的点积
    :param face1: 第一个面片，形状为 (n, 3)
    :param face2: 第二个面片，形状为 (n, 3)
    :return: 点积
    """
    # 将面片的每个点的坐标相乘
    multiplied = tf.multiply(face1, face2)
    
    # 对乘积后的结果求和，得到点积
    dot_product = tf.reduce_sum(multiplied, axis=1)
    
    return dot_product

# 示例数据
face1 = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], dtype=tf.float32)
face2 = tf.constant([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0]], dtype=tf.float32)

# 计算点积
dot_product = compute_dot_product(face1, face2)

# 打印结果
print(dot_product.numpy())