| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.nn |
| Modifier and Type | Method and Description |
|---|---|
<T extends Number> |
NnOps.topK(Operand<T> input,
Operand<Integer> k,
TopK.Options... options)
Builds an
TopK operation |
| Modifier and Type | Method and Description |
|---|---|
static TopK.Options |
TopK.sorted(Boolean sorted) |
TopK.Options |
TopK.Options.sorted(Boolean sorted) |
| Modifier and Type | Method and Description |
|---|---|
static <T extends Number> |
TopK.create(Scope scope,
Operand<T> input,
Operand<Integer> k,
TopK.Options... options)
Factory method to create a class wrapping a new TopK operation.
|
Copyright © 2015–2019. All rights reserved.