Review: MolTrans (2)

less than 1 minute read

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 개의 샘플 배치