Review: MolTrans (2)
embedding = nn.Embedding(10, 3)
an Embedding module containing 10 tensors of size 3
크기 3의 10 개의 텐서를 포함하는 Embedding 모듈
tensor( [ [ [..] , .. , [..] ] , [ [..] , .. , [..] ] ] )
input = torch.LongTensor([[1,2,4,5],[4,3,2,9]])
a batch of 2 samples of 4 indices each
각각 4 개의 인덱스로 구성된 2 개의 샘플 배치