jax.experimental.pallas.dot# jax.experimental.pallas.dot(a, b, trans_a=False, trans_b=False, allow_tf32=None, precision=None)[源代码]# 参数: trans_a (bool) trans_b (bool) allow_tf32 (bool | None | None)
