2025年12月24日
nn.Embedding
摘要： 我来通过几个具体例子讲解 nn.Embedding 的使用场景： 1. 基础示例：词嵌入 import torch import torch.nn as nn # 示例1：创建简单的词嵌入层 # 假设我们有5个词，每个词用3维向量表示 embedding = nn.Embedding(num_emb 阅读全文
posted @ 2025-12-24 21:58 saulstavo 阅读(6) 评论(0) 推荐(0)