Fonction torch.broadcast_shapes de PyTorch
Manuel de référence Pytorch torch
torch.broadcast_shapesC'est une fonction de PyTorch utilisée pour calculer le résultat de la diffusion de plusieurs formes. Elle renvoie la forme obtenue après diffusion des formes d'entrée sans réellement créer de tenseurs.
Définition de fonction
torch.broadcast_shapes(*shapes)
Exemple d'utilisation
Exemple
import torch
# Calculer le résultat de la diffusion de forme
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Sortie : (3, 4)
# Formes multiples
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Sortie : (3, 1, 5)
# Forme unique
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Sortie : (2, 3)
# Une erreur est signalée lorsque la diffusion n'est pas possible
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Erreur de diffusion :", e)
# Calculer le résultat de la diffusion de forme
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Sortie : (3, 4)
# Formes multiples
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Sortie : (3, 1, 5)
# Forme unique
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Sortie : (2, 3)
# Une erreur est signalée lorsque la diffusion n'est pas possible
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Erreur de diffusion :", e)
Autres extensions