ICode9

记得在代码的开始引入 import torch import torch.nn as nn 举个常用的例子 #以下代码为pytorch的python代码 embedding = nn.Embedding(10, 3) print(embedding.weight) input = torch.LongTensor([[0, 2, 0, 5]]) print(input) print(embedd