Function TopK

tensorflow/go/op/wrappers.go:18391–18408

Finds values and indices of the `k` largest elements for the last dimension. DEPRECATED at GraphDef version 7: Use TopKV2 instead If the input is a vector (rank-1), finds the `k` largest entries in the vector and outputs their values and indices as vectors. Thus `values[j]` is the `j`-th largest

(scope *Scope, input tf.Output, k int64, optional ...TopKAttr)

//
// Returns The `k` largest elements along each last dimensional slice.The indices of `values` within the last dimension of `input`.
func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices tf.Output) {
	if scope.Err() != nil {
		return
	}
	attrs := map[string]interface{}{"k": k}
	for _, a := range optional {
		a(attrs)
	}
	opspec := tf.OpSpec{
		Type: "TopK",
		Input: []tf.Input{
			input,
		},
		Attrs: attrs,
	}
	op := scope.AddOperation(opspec)
	return op.Output(0), op.Output(1)
}

// Generate the bucket boundaries for each feature based on accumulated summaries.
//