PyTorch
torch / torch
torch.outer¶
-
torch.
outer
(input, vec2, *, out=None) → Tensor¶ Outer product of
input
andvec2
. Ifinput
is a vector of size n andvec2
is a vector of size m , thenout
must be a matrix of size (n×m) .Note
This function does not broadcast.
- Parameters
- Keyword Arguments
out (Tensor, optional) – optional output matrix
Example:
>>> 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.]])
此页内容是否对您有帮助
感谢反馈!