MaxPoolGradWithArgmax.Options

public static class MaxPoolGradWithArgmax.Options

Optional attributes for MaxPoolGradWithArgmax

Public Methods

MaxPoolGradWithArgmax.Options

includeBatchInIndex(Boolean includeBatchInIndex)

Inherited Methods

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

Public Methods

public MaxPoolGradWithArgmax.Options includeBatchInIndex (Boolean includeBatchInIndex)

Parameters

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