private cudnnAddTensor ( |
||
handle | ||
alpha | double | |
biasDesc | ||
biasData | ManagedCuda.BasicTypes.CUdeviceptr | |
beta | double | |
srcDestDesc | ||
srcDestData | ManagedCuda.BasicTypes.CUdeviceptr | |
리턴 | cudnnStatus |
/* Tensor Bias addition : srcDest = alpha * bias + beta * srcDestDesc */ public void AddTensor(cudnnAddMode mode, double alpha, TensorDescriptor biasDesc, CudaDeviceVariable <double> biasData, double beta, TensorDescriptor srcDestDesc, CudaDeviceVariable <double> srcDestData ) { res = CudaDNNNativeMethods.cudnnAddTensor(_handle, mode, ref alpha, biasDesc.Desc, biasData.DevicePointer, ref beta, srcDestDesc.Desc, srcDestData.DevicePointer); Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnAddTensor", res)); if (res != cudnnStatus.Success) { throw new CudaDNNException(res); } }