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

Function BatchMatMul

tensorflow/go/op/wrappers.go:31583–31600  ·  view source on GitHub ↗

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)

Source from the content-addressed store, hash-verified

31581//
31582// Returns 3-D or higher with shape `[..., r_o, c_o]`
31583func 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.
31603type MatrixSolveLsAttr func(optionalAttr)

Callers 8

TEST_FFunction · 0.85
TEST_FFunction · 0.85
TEST_FFunction · 0.85
TEST_FFunction · 0.85
TestMatMulGradMethod · 0.85
MatMulGradHelperFunction · 0.85
TESTFunction · 0.85

Calls 4

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

Tested by 6

TEST_FFunction · 0.68
TEST_FFunction · 0.68
TEST_FFunction · 0.68
TEST_FFunction · 0.68
TestMatMulGradMethod · 0.68
TESTFunction · 0.68