public class cudnnSoftmaxMode extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_SOFTMAX_MODE_CHANNEL
compute the softmax over all C for each H, W, N
|
static int |
CUDNN_SOFTMAX_MODE_INSTANCE
compute the softmax over all C, H, W for each N
|
Modifier and Type | Method and Description |
---|---|
static String |
stringFor(int n)
Returns a string representation of the given constant
|
public static final int CUDNN_SOFTMAX_MODE_INSTANCE
public static final int CUDNN_SOFTMAX_MODE_CHANNEL
public static String stringFor(int n)
Copyright © 2020. All rights reserved.