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