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

Function ScatterNd

tensorflow/go/op/wrappers.go:33238–33250  ·  view source on GitHub ↗

Scatter `updates` into a new tensor according to `indices`. Creates a new tensor by applying sparse `updates` to individual values or slices within a tensor (initially zero for numeric, empty for string) of the given `shape` according to indices. This operator is the inverse of the `tf.gather_nd`

(scope *Scope, indices tf.Output, updates tf.Output, shape tf.Output)

Source from the content-addressed store, hash-verified

33236// Returns A new tensor with the given shape and updates applied according
33237// to the indices.
33238func ScatterNd(scope *Scope, indices tf.Output, updates tf.Output, shape tf.Output) (output tf.Output) {
33239 if scope.Err() != nil {
33240 return
33241 }
33242 opspec := tf.OpSpec{
33243 Type: "ScatterNd",
33244 Input: []tf.Input{
33245 indices, updates, shape,
33246 },
33247 }
33248 op := scope.AddOperation(opspec)
33249 return op.Output(0)
33250}
33251
33252// RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebugAttr is an optional argument to RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.
33253type RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebugAttr func(optionalAttr)

Callers 3

BuildGraphMethod · 0.85
TEST_FFunction · 0.85
GatherNdGradFunction · 0.85

Calls 3

ErrMethod · 0.45
AddOperationMethod · 0.45
OutputMethod · 0.45

Tested by 1

TEST_FFunction · 0.68