|
public GetConvolutionNdForwardOutputDim ( |
||
| inputTensorDesc | Handle to a previously initialized tensor descriptor. | |
| filterDesc | Handle to a previously initialized filter descriptor. | |
| nbDims | int | Dimension of the output tensor |
| tensorOuputDimA | int | Array of dimensions nbDims that contains on exit of this routine the sizes /// of the output tensor |
| return | void | |
public void GetConvolutionNdForwardOutputDim(TensorDescriptor inputTensorDesc,
FilterDescriptor filterDesc,
int nbDims,
int[] tensorOuputDimA
)
{
res = CudaDNNNativeMethods.cudnnGetConvolutionNdForwardOutputDim(_desc, inputTensorDesc.Desc, filterDesc.Desc, nbDims, tensorOuputDimA);
Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnGetConvolutionNdForwardOutputDim", res));
if (res != cudnnStatus.Success) throw new CudaDNNException(res);
}