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)
| 34470 | // *Note*: The default kernel implementation for MatMul on GPUs uses |
| 34471 | // cublas. |
| 34472 | func 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 | // |