std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cudnn

Function std::cuda_concat_channels

#include <std/ml/gpu_cudnn.h> source ยท definition

int cuda_concat_channels(
    struct CudaBuf a,
    unsigned long aChannels,
    struct CudaBuf c,
    unsigned long cChannels,
    struct CudaBuf out,
    unsigned long batch,
    unsigned long plane,
)

out[b] = (a[b], c[b]) along channels, for each of 'batch' images whose planes are 'plane' = height*width elements (concat_channels).