jnp.matmul 和 jnp.dot 都是用于矩阵乘法的函数，但它们在处理多维数组（即张量）时有不同的行为。以下是它们的区别和具体用法： jnp.dot 主要用于向量点积和矩阵乘法。对于一维数组，计算向量的点积。对于二维数组，计算标准的矩阵乘法。对于多维数组，按照最后一个维度与倒数第二个维度进行计算。 import jax.numpy as jnp# 向量点积a = jnp.array