Embedding快速理解维度变化

粗糙的解释维度

import torch
import torch.nn as nn
y=torch.randint(0,3,(10,3))
print(y)

定义一个10*3的矩阵，里面的值随机取0，1，2

m=nn.Embedding(3,5)
print(m.weight)

展示该Embedding里面值

print(m(y))

y中的值相当于索引，在Embedding中取对应的行

print(m(y).shape)

y经过m嵌入之后，维度变成了（10，3，5）.因为y是（10，3）的矩阵，每一行对应3个索引在Embedding中取对应的值。共10*3个索引，故嵌入处理之后，维度百变成了（10，3，5）