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, toString
public 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()
Operand
Inputs 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.