| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
<T,U extends Number,V extends Number> |
Ops.gather(Operand<T> params,
Operand<U> indices,
Operand<V> axis,
Gather.Options... options)
Builds an
Gather operation |
| Modifier and Type | Method and Description |
|---|---|
static Gather.Options |
Gather.batchDims(Long batchDims) |
Gather.Options |
Gather.Options.batchDims(Long batchDims) |
| Modifier and Type | Method and Description |
|---|---|
static <T,U extends Number,V extends Number> |
Gather.create(Scope scope,
Operand<T> params,
Operand<U> indices,
Operand<V> axis,
Gather.Options... options)
Factory method to create a class wrapping a new Gather operation.
|
Copyright © 2015–2019. All rights reserved.