~/jendhub

def bmm

bendtorch@1.3.0.0 · tensor.bend:686

import bendtorch@1.3.0.0/tensor.bend as Tensor
import Base

686def bmm(+bs: Nat, +n: Nat, +k: Nat, +m: Nat, x: Tensor<[bs, n, k]>, y: Tensor<[bs, k, m]>) -> T3<[bs, n, k], [bs, k, m], [bs, n, m]>:
687  match x y:
688    case Tensor{dx} Tensor{dy}:
689      t3_of([bs, n, k], [bs, k, m], [bs, n, m], bmm_l(Nat.mul(bs, Nat.mul(n, m)), 0, Geo{u(n), u(m), k, u(Nat.mul(n, k)), u(k), 1, u(Nat.mul(k, m)), u(m), 1}, dx, dy, new(Nat.mul(bs, Nat.mul(n, m)))))

Uses (10)

Used by (0)

No other definition uses this definition.