Función PyTorch torch.broadcast_shapes

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.broadcast_shapesEs una función en PyTorch que se utiliza para calcular el resultado de la transmisión (broadcast) de múltiples formas. Devuelve la forma resultante después de transmitir las formas de entrada, sin crear realmente tensores.

Definición de la función

torch.broadcast_shapes(*shapes)

Ejemplo de uso

Ejemplo

import torch

# Calcular el resultado de broadcast de las formas
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Salida: (3, 4)

# Múltiples formas
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Salida: (3, 1, 5)

# Forma única
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Salida: (2, 3)

# Si no se puede transmitir, se produce un error
try:
    result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
    print("Error de broadcast:", e)

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones