gpuDeltaX¶
-
ConvolutionalLayer.
gpuDeltaX
()¶ GPUによるδxの計算
ソース¶
gpuDeltaX() {
var prev_layer = this.prevLayer;
var param_id = "ConvolutionalLayer-delta-X:" + this.filterSize + ":" + prev_layer.numChannels + ":" + this.numChannels + ":" + this.numRows + ":" + this.numCols + ":" + miniBatchSize;
if (this.params[param_id] == undefined) {
var shader_src = Shaders.ConvolutionalLayer_DeltaX
.replace(/miniBatchSize/g, miniBatchSize.toString() + "u")
.replace(/numChannels/g, this.numChannels.toString() + "u")
.replace(/prevNumChannels/g, prev_layer.numChannels.toString() + "u")
.replace(/prevNumRows/g, prev_layer.numRows.toString() + "u")
.replace(/prevNumCols/g, prev_layer.numCols.toString() + "u")
.replace(/numRows/g, this.numRows.toString() + "u")
.replace(/numCols/g, this.numCols.toString() + "u")
.replace(/filterSize/g, this.filterSize.toString() + "u");
this.params[param_id] = {
id : param_id,
vertexShader: shader_src,
args : {
"zero": new Float32Array(this.deltaX.dt.length),
"weight": makeTextureInfo(WebGL2, "float", new ArrayView(this.numChannels * prev_layer.numChannels, this.filterSize, this.filterSize)),
"delta_z": makeTextureInfo(WebGL2, "float", new ArrayView(miniBatchSize * this.numChannels, this.numRows, this.numCols)),
"delta_x": this.deltaX.dt
}
};
}
var param = this.params[param_id];
param.args["weight"].value = this.weight.dt;
param.args["delta_z"].value = this.deltaZ.dt;
WebGL2.compute(param);
}