//clika-runtime/io.clika.runtime/Ops/tensordot

tensordot

[common]
fun tensordot(a: Tensor, b: Tensor, dimsA: LongArray, dimsB: LongArray): Tensor

tensordot(a: Tensor, b: Tensor, dimsA: LongArray, dimsB: LongArray): the tensordot operator. Named-axis contraction: sums a over dims_a against b over dims_b, pairwise. dims_a[i] on a contracts with dims_b[i] on b (equal list lengths; per-pair sizes must agree). The output is a's non-contracted dims followed by b's.