-
-
Notifications
You must be signed in to change notification settings - Fork 751
Closed
Labels
Description
I want to use adaptive_avg_pool2d
function to transform input tensor to output tensor:
output = adaptive_avg_pool2d(input, new LongArrayRef(new long[] { 7, 7 }, 2));
The input variable's shape is [64, 512, 7, 7]
, but the output's shape is [64, 512, 0, 7]
.
The correct output shape should be [64, 512, 7, 7]