Поэтапное произведение PyTorch векторов / матриц / тензоров

Question

Как в PyTorch я могу получить поэлементное произведение двух векторов / матриц / тензоров?

Для googlers этот продукт также известен как:

• продукт Адамара
• продукт Шура
• Вводное произведение

Answer 1

При наличии двух тензоров A и B вы можете использовать либо:

• A * B
• torch.mul(A, B)
• A.mul(B)

Примечание : для умножения матриц вы хотите использовать A @ B, что эквивалентно torch.matmul().