MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MatMul

Function MatMul

tensorflow/go/op/wrappers.go:34472–34489  ·  view source on GitHub ↗

Multiply the matrix "a" by the matrix "b". The inputs must be two-dimensional matrices and the inner dimension of "a" (after being transposed if transpose_a is true) must match the outer dimension of "b" (after being transposed if transposed_b is true). *Note*: The default kernel implementation fo

(scope *Scope, a tf.Output, b tf.Output, optional ...MatMulAttr)

Source from the content-addressed store, hash-verified

34470// *Note*: The default kernel implementation for MatMul on GPUs uses
34471// cublas.
34472func MatMul(scope *Scope, a tf.Output, b tf.Output, optional ...MatMulAttr) (product tf.Output) {
34473 if scope.Err() != nil {
34474 return
34475 }
34476 attrs := map[string]interface{}{}
34477 for _, a := range optional {
34478 a(attrs)
34479 }
34480 opspec := tf.OpSpec{
34481 Type: "MatMul",
34482 Input: []tf.Input{
34483 a, b,
34484 },
34485 Attrs: attrs,
34486 }
34487 op := scope.AddOperation(opspec)
34488 return op.Output(0)
34489}
34490
34491// Elementwise computes the bitwise left-shift of `x` and `y`.
34492//

Callers 6

TestDeviceFunction · 0.70
ExampleFunction · 0.70
CreateTestGraphFunction · 0.50
TestIgnoreOpsMethod · 0.50
TestQuantizeMatMulMethod · 0.50

Calls 4

aClass · 0.85
ErrMethod · 0.45
AddOperationMethod · 0.45
OutputMethod · 0.45

Tested by 6

TestDeviceFunction · 0.56
ExampleFunction · 0.56
CreateTestGraphFunction · 0.40
TestIgnoreOpsMethod · 0.40
TestQuantizeMatMulMethod · 0.40