首页 > TAG信息列表 > 0.1371
pytorch中nn.Embedding()的用法
记得在代码的开始引入 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(embeddpytorch中nn.Embedding()的用法
记得在代码的开始引入 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