Communauté RSS DEV

MaxPool3d() dans PyTorch

MaxPool3d peut calculer le tenseur 4D ou 5D d'un ou plusieurs valeurs calculées par le pooling max 3D à partir du tenseur 4D ou 5D d'un ou plusieurs éléments. Les arguments de taille de noyau, de stride, de padding et de dilation doivent être des entiers supérieurs ou égaux à 1, tandis que les arguments return_indices et ceil_mode sont des drapeaux booléens. Par défaut, l'attribut requires_grad du tenseur est faux et reste faux après utilisation de MaxPool3d. Lorsque return_indices est faux, MaxPool3d renvoie uniquement le tenseur de sortie. Lorsque return_indices est vrai, il renvoie également un tenseur contenant les indices des valeurs maximales dans le tenseur d'entrée. Augmenter la taille du noyau augmente le champ de réception et réduit la taille de sortie, tandis qu'augmenter le stride réduit la taille de sortie et introduit des lacunes dans le champ de réception. Le padding peut être utilisé pour ajouter des zéros autour du tenseur d'entrée pour contrôler la taille de sortie et éviter la perte d'information. La dilation augmente l'espace entre les éléments du noyau sans modifier sa taille, permettant des champs de réception plus grands sans réduire la taille de sortie. Le mode ceil garantit que la taille de sortie est toujours arrondie à l'entier supérieur le plus proche, ce qui peut être utile pour certaines applications.
favicon
dev.to
MaxPool3d() in PyTorch