Spec-Zone.ru › PyTorch 2

torch.outer

torch.outer(input, vec2, *, out=None) → Tensor

Внешнее произведение input и vec2. Если input является вектором размера nn и vec2 является вектором размера mm, то out должен быть матрицей размера (n×m)(n \times m).

Примечание

Эта функция не распространяется.

Параметры
  • input (Tensor) – Вектор входных данных размерности 1
  • vec2 (Tensor) – Вектор входных данных размерности 1
Аргументы по ключевым словам

out (Tensor, необязательно) – необязательная выходная матрица

Пример:

>>> v1 = torch.arange(1., 5.)
>>> v2 = torch.arange(1., 4.)
>>> torch.outer(v1, v2)
tensor([[  1.,   2.,   3.],
        [  2.,   4.,   6.],
        [  3.,   6.,   9.],
        [  4.,   8.,  12.]])

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.outer.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API