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);
}