예제 #1
0
        public CudnnConvolutionTensorDim GetOutputTensor(CudnnConvolutionPath path)
        {
            Contract.Ensures(Contract.Result <CudnnConvolutionTensorDim>() != null);

            ThrowIfNotInitialized();

            int n = 0, c = 0, h = 0, w = 0;

            CudnnContext.Invoke(() => CudnnNativeMethods.cudnnGetOutputTensor4dDim(this.Handle, path, out n, out c, out h, out w));

            return(new CudnnConvolutionTensorDim(path, n, c, h, w));
        }