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 ...BatchMatMulV2Attr)
| 31785 | // |
| 31786 | // Returns 3-D or higher with shape `[..., r_o, c_o]` |
| 31787 | func BatchMatMulV2(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulV2Attr) (output tf.Output) { |
| 31788 | if scope.Err() != nil { |
| 31789 | return |
| 31790 | } |
| 31791 | attrs := map[string]interface{}{} |
| 31792 | for _, a := range optional { |
| 31793 | a(attrs) |
| 31794 | } |
| 31795 | opspec := tf.OpSpec{ |
| 31796 | Type: "BatchMatMulV2", |
| 31797 | Input: []tf.Input{ |
| 31798 | x, y, |
| 31799 | }, |
| 31800 | Attrs: attrs, |
| 31801 | } |
| 31802 | op := scope.AddOperation(opspec) |
| 31803 | return op.Output(0) |
| 31804 | } |
| 31805 | |
| 31806 | // CastAttr is an optional argument to Cast. |
| 31807 | type CastAttr func(optionalAttr) |