We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a54be74 commit 80f4dc6Copy full SHA for 80f4dc6
tfjs-layers/src/layers/preprocessing/image_resizing.ts
@@ -68,7 +68,8 @@ export class Resizing extends Layer {
68
69
computeOutputShape(inputShape: Shape|Shape[]): Shape|Shape[] {
70
inputShape = getExactlyOneShape(inputShape);
71
- const numChannels = inputShape.slice(inputShape.length - 1)[0];
+ // inputShape.slice(inputShape.length - 1)[0];
72
+ const numChannels = inputShape.slice()[2];
73
const outputShape = [];
74
outputShape.push(this.height);
75
outputShape.push(this.width);
0 commit comments