Matmul
public final class Matmul : Model
Matrix-multiplication over two inputs.
-
Declaration
Swift
public init(transposeA: (Int, Int) = (0, 0), transposeB: (Int, Int) = (0, 0), name: String = "")
-
Declaration
Swift
public func callAsFunction<T: DynamicGraph.TensorGroup>( _ left: T, _ right: T, streamContext: StreamContext? = nil ) -> T