Method RoiAlign
RoiAlign(TensorFloat, TensorFloat, TensorInt, RoiPoolingMode, int, int, int, float)
Calculates an output tensor by pooling the input tensor across each region of interest given by the rois
tensor.
Declaration
public TensorFloat RoiAlign(TensorFloat X, TensorFloat Rois, TensorInt Indices, RoiPoolingMode mode, int outputHeight, int outputWidth, int samplingRatio, float spatialScale)
Parameters
Type | Name | Description |
---|---|---|
TensorFloat | X | The input tensor. |
TensorFloat | Rois | The region of interest input tensor. |
TensorInt | Indices | The indices input tensor. |
RoiPoolingMode | mode | The pooling mode of the operation as an |
int | outputHeight | The height of the output tensor. |
int | outputWidth | The width of the output tensor. |
int | samplingRatio | The number of sampling points in the interpolation grid used to compute the output value of each pooled output bin. |
float | spatialScale | The multiplicative spatial scale factor used to translate coordinates from their input spatial scale to the scale used when pooling. |
Returns
Type | Description |
---|---|
TensorFloat | The computed output tensor. |