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

Function BatchMatMulV2

tensorflow/go/op/wrappers.go:31787–31804  ·  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 ...BatchMatMulV2Attr)

Source from the content-addressed store, hash-verified

31785//
31786// Returns 3-D or higher with shape `[..., r_o, c_o]`
31787func 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.
31807type CastAttr func(optionalAttr)

Callers 3

TEST_FFunction · 0.85
VerifyFusedMethod · 0.85
TEST_FFunction · 0.85

Calls 4

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

Tested by 3

TEST_FFunction · 0.68
VerifyFusedMethod · 0.68
TEST_FFunction · 0.68