|
public GetPoolingNdForwardOutputDim ( |
||
| inputTensorDesc | Handle to the previously initialized input tensor descriptor. | |
| nbDims | int | Number of dimensions in which pooling is to be applied. |
| outputTensorDimA | int | Array of nbDims output dimensions |
| return | void | |
public void GetPoolingNdForwardOutputDim(TensorDescriptor inputTensorDesc,
int nbDims,
int[] outputTensorDimA)
{
res = CudaDNNNativeMethods.cudnnGetPoolingNdForwardOutputDim(_desc, inputTensorDesc.Desc, nbDims, outputTensorDimA);
Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnGetPoolingNdForwardOutputDim", res));
if (res != cudnnStatus.Success) throw new CudaDNNException(res);
}