pub unsafe fn set_tensornd_descriptor(
    tensor_desc: cudnnTensorDescriptor_t,
    data_type: cudnnDataType_t,
    num_dims: c_int,
    dims: *const c_int,
    strides: *const c_int
) -> Result<(), CudnnError>
Expand description

Sets data on a tensor descriptor. See nvidia docs

Safety

tensor_desc must have been created with create_tensor_descriptor, and NOT freed by destroy_tensor_descriptor