T
- data type for output()
output@Operator(group="nn") public final class Dilation2d<T extends Number> extends PrimitiveOp implements Operand<T>
The `input` tensor has shape `[batch, in_height, in_width, depth]` and the `filter` tensor has shape `[filter_height, filter_width, depth]`, i.e., each input channel is processed independently of the others with its own structuring function. The `output` tensor has shape `[batch, out_height, out_width, depth]`. The spatial dimensions of the output tensor depend on the `padding` algorithm. We currently only support the default "NHWC" `data_format`.
In detail, the grayscale morphological 2-D dilation is the max-sum correlation (for consistency with `conv2d`, we use unmirrored filters):
output[b, y, x, c] = max_{dy, dx} input[b, strides[1] * y + rates[1] * dy, strides[2] * x + rates[2] * dx, c] + filter[dy, dx, c]
Max-pooling is a special case when the filter has size equal to the pooling kernel size and contains all zeros.
Note on duality: The dilation of `input` by the `filter` is equal to the negation of the erosion of `-input` by the reflected `filter`.
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number> |
create(Scope scope,
Operand<T> input,
Operand<T> filter,
List<Long> strides,
List<Long> rates,
String padding)
Factory method to create a class wrapping a new Dilation2d operation.
|
Output<T> |
output()
4-D with shape `[batch, out_height, out_width, depth]`.
|
equals, hashCode, op, toString
public static <T extends Number> Dilation2d<T> create(Scope scope, Operand<T> input, Operand<T> filter, List<Long> strides, List<Long> rates, String padding)
scope
- current scopeinput
- 4-D with shape `[batch, in_height, in_width, depth]`.filter
- 3-D with shape `[filter_height, filter_width, depth]`.strides
- The stride of the sliding window for each dimension of the input
tensor. Must be: `[1, stride_height, stride_width, 1]`.rates
- The input stride for atrous morphological dilation. Must be:
`[1, rate_height, rate_width, 1]`.padding
- The type of padding algorithm to use.public Output<T> 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<T extends Number>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.