더북(TheBook)

permute가 적용되면 다음 결과처럼 행과 열이 바뀌어 출력되는 것을 확인할 수 있습니다.

tensor([[[0.4027, 0.3268, 0.8037],
         [0.8018, 0.3750, 0.0183],
         [0.2184, 0.3658, 0.8037]],

        [[0.1167, 0.4351, 0.8115],
         [0.5717, 0.0638, 0.2578],
         [0.6361, 0.2452, 0.7153]]])
tensor([[[0.4027, 0.1167],
         [0.8018, 0.5717],
         [0.2184, 0.6361]],

        [[0.3268, 0.4351],
         [0.3750, 0.0638],
         [0.3658, 0.2452]],
        
        [[0.8037, 0.8115],
         [0.0183, 0.2578],
         [0.8037, 0.7153]]])

다음은 permute를 적용하여 차원이 변환된 결과입니다.

torch.Size([33, 13, 768])

쪼개진 단어에 대한 벡터를 생성합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.