@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
12 lines • 543 B
JavaScript
export function outputDimSize(inSize, kernel, headPad, tailPad, dilation, stride) {
const dkernel = dilation * (kernel - 1) + 1;
return Math.floor((inSize + headPad + tailPad - dkernel) / stride + 1);
}
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=conv.js.map