public static class MaxPoolGradGradWithArgmax.Options extends Object
MaxPoolGradGradWithArgmax
Modifier and Type | Method and Description |
---|---|
MaxPoolGradGradWithArgmax.Options |
includeBatchInIndex(Boolean includeBatchInIndex) |
public MaxPoolGradGradWithArgmax.Options includeBatchInIndex(Boolean includeBatchInIndex)
includeBatchInIndex
- Whether to include batch dimension in flattened index of `argmax`.Copyright © 2022. All rights reserved.