gpuDeltaX

FullyConnectedLayer.gpuDeltaX()

GPUによるδxの計算

ソース

gpuDeltaX(){
    var vertex_shader = Shaders.FullyConnectedLayer_DeltaX;

    var param_id = "Fully-Connected-Layer-gpu-delta-X," + miniBatchSize + "," + this.prevLayer.unitSize + "," + this.unitSize;
    if (this.params[param_id] == undefined){

        this.params[param_id] = {
            id : param_id,
            vertexShader: vertex_shader,
            args : {
                "zero": new Float32Array(miniBatchSize * this.prevLayer.unitSize),
                "W": makeTextureInfo(WebGL2, "float", this.weight),
                "deltaZ": makeTextureInfo(WebGL2, "float", this.deltaZ),
                "deltaX" : this.deltaX.dt
            }
        };
    }

    var param = this.params[param_id];
    param.args["deltaZ"].value = this.deltaZ.dt;

    WebGL2.compute(param);
}