Library/ManagedCuda/CudaDNN/CudaDNNNativeMethods.cs
/// <summary>
/// This function generates a grid of coordinates in the input tensor corresponding to each pixel from the output tensor.
/// </summary>
/// <param name="handle">Handle to a previously created cuDNN context.</param>
/// <param name="stDesc">Previously created spatial transformer descriptor object.</param>
/// <param name="theta">Affine transformation matrix. It should be of size n*2*3 for a 2d transformation, where n is the number of images specified in stDesc.</param>
/// <param name="grid">A grid of coordinates. It is of size n*h*w*2 for a 2d transformation, where n, h, w is specified in stDesc. In the 4th dimension, the first coordinate is x, and the second coordinate is y.</param>
This function generates a grid of coordinates in the input tensor corresponding to each pixel from the output tensor.
This function generates a grid of coordinates in the input tensor corresponding to each pixel from the output tensor .
[DllImport(CUDNN_API_DLL_NAME)]
        public static extern cudnnStatus cudnnSpatialTfGridGeneratorForward(
                                         cudnnHandle handle,
                                         cudnnSpatialTransformerDescriptor stDesc,
                                         CUdeviceptr theta,
                                         CUdeviceptr grid);
[ DllImport ( CUDNN_API_DLL_NAME ) ] public static extern cudnnStatus cudnnSpatialTfGridGeneratorForward ( cudnnHandle handle , cudnnSpatialTransformerDescriptor stDesc , CUdeviceptr theta , CUdeviceptr grid ) ;
This function generates a grid of coordinates in the input tensor corresponding to each pixel from the output tensor.
This function generates a grid of coordinates in the input tensor corresponding to each pixel from the output tensor .
handle
cudnnHandle
Handle to a previously created cuDNN context.
Handle to a previously created cuDNN context .
stDesc
cudnnSpatialTransformerDescriptor
Previously created spatial transformer descriptor object.
Previously created spatial transformer descriptor object .
theta
CUdeviceptr
Affine transformation matrix. It should be of size n*2*3 for a 2d transformation, where n is the number of images specified in stDesc.
Affine transformation matrix . It should be of size n * 2 * 3 for a 2d transformation where n is the number of images specified in stDesc .
grid
CUdeviceptr
A grid of coordinates. It is of size n*h*w*2 for a 2d transformation, where n, h, w is specified in stDesc. In the 4th dimension, the first coordinate is x, and the second coordinate is y.
A grid of coordinates . It is of size n * h * w * 2 for a 2d transformation where n h w is specified in stDesc . In the 4th dimension the first coordinate is x and the second coordinate is y .