La façon de remplir avec des zéros d'un tenseur, le long de certains axes (Python)
Je voudrais ajouter un numpy tenseur avec 0 le long de l'axe choisi.
Par exemple, j'ai du tenseur de r
la forme de la (4,3,2)
mais je suis seulement intéressé par un rembourrage seuls les deux derniers axe (qui est, que le pavé de la matrice). Est-il possible de le faire avec une ligne de code python?
Vous devez vous connecter pour publier un commentaire.
Vous pouvez utiliser
np.pad()
:Cette fonction serait pad à la fin de certains axes.
Si vous souhaitez pad à la fois côté, juste le modifier.
exemple: