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