Умножим все элементы тензора PyTorch

Question

Я хотел сделать что-то вроде этого вопрос в PyTorch, то есть умножить все элементы друг на друга, поддерживая определенную ось постоянной. Есть ли в PyTorch какая-либо предопределенная функция для этого?

Answer 1

Да. torch.prod . Используйте параметр dim, чтобы указать, по какой оси вы хотите рассчитать произведение.

x = torch.randn((2, 2))
print(x)
print(torch.prod(x, 0)) # product along 0th axis

Это печатает

tensor([[-0.3661, 1.0693],
           [0.5144, 1.3489]])
tensor([-0.1883, 1.4424])

Answer 2

Если вы хотите сделать:
умножение матрицы, вы можете использовать torch.matmul
умножить матрицу на скаляр (или тензор со скалярами), вы можете использовать torch.multiply

Документы тензор очень обширны по этому вопросу ... вы должны взглянуть