2025年1月7日
转：torch.nn.Embedding函数用法图解
摘要： 【python函数】torch.nn.Embedding函数用法图解-CSDN博客 import torch import torch.nn as nn embedding = nn.Embedding(10, 3) # 10表示num_embeddings, 3表示embedding_dim。用标 阅读全文
posted @ 2025-01-07 20:55 Picassooo 阅读(83) 评论(0) 推荐(0)