org.tensorflow.op.nn

Class MaxPoolGradWithArgmax.Options

java.lang.Object

org.tensorflow.op.nn.MaxPoolGradWithArgmax.Options

Enclosing class:

MaxPoolGradWithArgmax<T extends Number>

public static class MaxPoolGradWithArgmax.Options
extends Object

Optional attributes for MaxPoolGradWithArgmax

Method Summary

Modifier and Type	Method and Description
MaxPoolGradWithArgmax.Options	includeBatchInIndex(Boolean includeBatchInIndex)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

includeBatchInIndex

public MaxPoolGradWithArgmax.Options includeBatchInIndex(Boolean includeBatchInIndex)

Parameters:

includeBatchInIndex - Whether to include batch dimension in flattened index of `argmax`.