Multiplies slices of two tensors in batches. Multiplies all slices of `Tensor` `x` and `y` (each slice can be viewed as an element of a batch), and arranges the individual results in a single output tensor of the same batch size. Each of the individual slices can optionally be adjointed (to adjoint
(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr)
| 31581 | // |
| 31582 | // Returns 3-D or higher with shape `[..., r_o, c_o]` |
| 31583 | func BatchMatMul(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr) (output tf.Output) { |
| 31584 | if scope.Err() != nil { |
| 31585 | return |
| 31586 | } |
| 31587 | attrs := map[string]interface{}{} |
| 31588 | for _, a := range optional { |
| 31589 | a(attrs) |
| 31590 | } |
| 31591 | opspec := tf.OpSpec{ |
| 31592 | Type: "BatchMatMul", |
| 31593 | Input: []tf.Input{ |
| 31594 | x, y, |
| 31595 | }, |
| 31596 | Attrs: attrs, |
| 31597 | } |
| 31598 | op := scope.AddOperation(opspec) |
| 31599 | return op.Output(0) |
| 31600 | } |
| 31601 | |
| 31602 | // MatrixSolveLsAttr is an optional argument to MatrixSolveLs. |
| 31603 | type MatrixSolveLsAttr func(optionalAttr) |