Función PyTorch torch.bmm

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

torch.bmmEs una función en PyTorch que se utiliza para realizar multiplicación de matrices por lotes.

Definición de la función

torch.bmm(input, mat2, out)

Ejemplo de uso

Ejemplo

import torch

# Multiplicación de matrices por lotes
batch_a = torch.randn(10, 3, 4)
batch_b = torch.randn(10, 4, 5)

result = torch.bmm(batch_a, batch_b)

print("Forma del resultado por lotes:", result.shape)

El resultado de salida es:

批量结果形状: torch.Size([10, 3, 5])

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

Otras extensiones