Calcule une convolution en profondeur 2D à partir des tenseurs « entrée » et « filtre » 4-D.
Étant donné un tenseur d'entrée de forme `[batch, in_height, in_width, in_channels]` et un tenseur de filtre/noyau de forme `[filter_height, filter_width, in_channels, channel_multiplier]`, contenant des filtres convolutifs `in_channels` de profondeur 1, `deepwise_conv2d` applique un filtre différent à chaque canal d'entrée (en passant de 1 canal aux canaux `channel_multiplier` pour chacun), puis concatène les résultats ensemble. Ainsi, la sortie a des canaux `in_channels * channel_multiplier`.
for k in 0..in_channels-1
for q in 0..channel_multiplier-1
output[b, i, j, k * channel_multiplier + q] =
sum_{di, dj
input[b, strides[1] * i + di, strides[2] * j + dj, k] *
filter[di, dj, k, q]
}
Classes imbriquées
classe | DepthwiseConv2dNative.Options | Attributs facultatifs pour DepthwiseConv2dNative |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique <T étend TNumber > DepthwiseConv2dNative <T> | créer (portée de portée, entrée d'opérande <T>, filtre d'opérande <T>, foulées de liste <Long>, remplissage de chaîne, options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération DepthwiseConv2dNative. |
statique DepthwiseConv2dNative.Options | dataFormat (Chaîne dataFormat) |
statique DepthwiseConv2dNative.Options | dilatations (List<Long> dilatations) |
statique DepthwiseConv2dNative.Options | explicitPaddings (List<Long> explicitPaddings) |
Sortie <T> | sortir () |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static DepthwiseConv2dNative <T> créer (portée de portée , entrée d'opérande <T>, filtre d'opérande <T>, foulées de liste <Long>, remplissage de chaîne, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération DepthwiseConv2dNative.
Paramètres
portée | portée actuelle |
---|---|
foulées | 1-D de longueur 4. La foulée de la fenêtre glissante pour chaque dimension de « entrée ». |
rembourrage | Le type d’algorithme de remplissage à utiliser. |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de DepthwiseConv2dNative
public statique DepthwiseConv2dNative.Options dataFormat (String dataFormat)
Paramètres
format de données | Spécifiez le format de données des données d'entrée et de sortie. Avec le format par défaut « NHWC », les données sont stockées dans l'ordre : [lot, hauteur, largeur, canaux]. Alternativement, le format pourrait être « NCHW », l'ordre de stockage des données étant : [lot, canaux, hauteur, largeur]. |
---|
dilatations statiques publiques DepthwiseConv2dNative.Options (dilatations List<Long>)
Paramètres
dilatations | Tenseur 1-D de longueur 4. Le facteur de dilatation pour chaque dimension de « entrée ». Si défini sur k > 1, il y aura k-1 cellules ignorées entre chaque élément de filtre sur cette dimension. L'ordre des dimensions est déterminé par la valeur de `data_format`, voir ci-dessus pour plus de détails. Les dilatations dans les dimensions du lot et de la profondeur doivent être de 1. |
---|