U - data type for output() output@Operator public final class StridedSliceGrad<U> extends PrimitiveOp implements Operand<U>
Since `StridedSlice` cuts out pieces of its `input` which is size `shape`, its gradient will have the same shape (which is passed here as `shape`). The gradient will be zero in any element that the slice does not select.
Arguments are the same as StridedSliceGrad with the exception that `dy` is the input gradient to be propagated and `shape` is the shape of `StridedSlice`'s `input`.
| Modifier and Type | Class and Description |
|---|---|
static class |
StridedSliceGrad.Options
Optional attributes for
StridedSliceGrad |
operation| Modifier and Type | Method and Description |
|---|---|
Output<U> |
asOutput()
Returns the symbolic handle of a tensor.
|
static StridedSliceGrad.Options |
beginMask(Long beginMask) |
static <U,T extends Number> |
create(Scope scope,
Operand<T> shape,
Operand<T> begin,
Operand<T> end,
Operand<T> strides,
Operand<U> dy,
StridedSliceGrad.Options... options)
Factory method to create a class wrapping a new StridedSliceGrad operation.
|
static StridedSliceGrad.Options |
ellipsisMask(Long ellipsisMask) |
static StridedSliceGrad.Options |
endMask(Long endMask) |
static StridedSliceGrad.Options |
newAxisMask(Long newAxisMask) |
Output<U> |
output() |
static StridedSliceGrad.Options |
shrinkAxisMask(Long shrinkAxisMask) |
equals, hashCode, op, toStringpublic static <U,T extends Number> StridedSliceGrad<U> create(Scope scope, Operand<T> shape, Operand<T> begin, Operand<T> end, Operand<T> strides, Operand<U> dy, StridedSliceGrad.Options... options)
scope - current scopeshape - begin - end - strides - dy - options - carries optional attributes valuespublic static StridedSliceGrad.Options beginMask(Long beginMask)
beginMask - public static StridedSliceGrad.Options endMask(Long endMask)
endMask - public static StridedSliceGrad.Options ellipsisMask(Long ellipsisMask)
ellipsisMask - public static StridedSliceGrad.Options newAxisMask(Long newAxisMask)
newAxisMask - public static StridedSliceGrad.Options shrinkAxisMask(Long shrinkAxisMask)
shrinkAxisMask - public Output<U> asOutput()
OperandInputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput in interface Operand<U>OperationBuilder.addInput(Output)Copyright © 2022. All rights reserved.