@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
12 lines • 550 B
JavaScript
export function outputDimSize(inSize, kernel, headPad, tailPad, dilation, stride) {
const kernelSize = dilation * (kernel - 1) + 1;
return stride * (inSize - 1) + headPad + tailPad - kernelSize + 2;
}
export function outputDimsSize(inSizes, kernels, headPads, tailPads, dilations, strides) {
const result = [];
for (let i = 0; i < inSizes.length; i++) {
result.push(outputDimSize(inSizes[i], kernels[i], headPads[i], tailPads[i], dilations[i], strides[i]));
}
return result;
}
//# sourceMappingURL=convTranspose.js.map