Matmul

public final class Matmul : Model

Matrix-multiplication over two inputs.


                    
                    
                    init(transposeA:transposeB:flags:name:)

Swift

public init(
  transposeA: (Int, Int) = (0, 0), transposeB: (Int, Int) = (0, 0),
  flags: Functional.GEMMFlag = [], name: String = ""
)


                    
                    
                    callAsFunction(_:_:streamContext:)

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup>(
  _ left: T, _ right: T, streamContext: StreamContext? = nil
) -> T