org.tensorflow.op.nn

Class MaxPoolWithArgmax.Options

java.lang.Object

org.tensorflow.op.nn.MaxPoolWithArgmax.Options

Enclosing class:

MaxPoolWithArgmax<T extends Number,U extends Number>

public static class MaxPoolWithArgmax.Options
extends Object

Optional attributes for MaxPoolWithArgmax

Method Summary

Modifier and Type	Method and Description
MaxPoolWithArgmax.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 MaxPoolWithArgmax.Options includeBatchInIndex(Boolean includeBatchInIndex)

Parameters:

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