def dotted

bendtorch-core@0.1.0.0 · src/tensor/matmul.bend:144

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

144def dotted(-left: Shape.Shape, -right: Shape.Shape, -output: Shape.Shape,
145  rec: Core.Tensor<left> -> Core.Tensor<right> -> Core.Tensor<output> ->
146    Result<&2, &1, Error, Core.Tensor<left> & Core.Tensor<right> & Core.Tensor<output>>,
147  out: Core.Tensor<output>, index: U32,
148  result: Result<&2, &1, Error, Core.Tensor<left> & Core.Tensor<right> & F32>) ->
149  Result<&2, &1, Error, Core.Tensor<left> & Core.Tensor<right> & Core.Tensor<output>>:
150  match result:
151    case Fail{error}:
152      Fail{error}
153    case Done{values}:
154      (a, b, value) = values
155      wrote(left, right, output, rec, a, b, Core.write_flat(output, out, index, value))