[Pytorch] 행렬곱


[Pytorch] 행렬곱

행렬텐서에서 행렬은 랭크(차원) 2 텐서와 같다만약 행렬 A와 행렬 B가 있다면 행렬곱을 하기 위해서는 행렬 A의 열 수와 B의 행 수가 같아야 한다.따라서 A가 2*3 행렬이고 B가 3*4 행렬이라면 연산이 가능하지만 A가 3*4행렬이고 B가 2*5 행렬이라면 연산이 불가능하다.행렬곱의 연산 결과는 행렬 A의 행 수와 행렬 B의 열 수와 같다.행렬곱을 진행하기 전 아래와 같이 텐서들을 선언한다.a는 torch.randn()을 이용해 무작위 숫자로 이루어진 텐서를 생성하였고 b는 직접 실수형 원소를 넣어 생성하였다.행렬곱torch.mm(input, mat2, *, out=None)input은 행렬곱을 수행할 첫 번째 텐서, mat2는 행렬곱을 수행할 두 번째 텐서를 뜻한..........

[Pytorch] 행렬곱에 대한 요약내용입니다.

자세한 내용은 아래에 원문링크를 확인해주시기 바랍니다.



원문링크 : [Pytorch] 행렬곱