Function Mult

src/core/tensor/tensor.cc:1616–1627

}

Tensor Mult(const Tensor &A, const Tensor &B) {
  auto A_ = Broadcast(A, B.shape(), 2);
  auto B_ = Broadcast(B, A.shape(), 2);

  Shape s = A_.shape();
  s.pop_back();
  s.push_back(B.shape(B.nDim() - 1));

  Tensor out(s, A.device(), A.data_type());
  Mult(A_, B_, &out);
  return out;
}

void Mult(const Tensor &A, const Tensor &B, Tensor *out) {
  Mult(1.0f, A, B, 0.0f, out);