nn.Embedding()个人记录

 维度

import torch.nn as nn
 
embedding = nn.Embedding(num_embeddings = 10, embedding_dim = 256)

nn.Embedding()随机产生一个权重矩阵weight,维度为(num_embeddings, embedding_dim) 

输入维度(batch_size, Seq_len)

输出维度(batch_size,Seq_len,embedding_dim)

举例

 

参考&转载:

pytorch复习笔记--nn.Embedding()的用法-CSDN博客 

相关推荐

  1. cnn报错个人记录

    2023-12-24 18:22:02       39 阅读

最近更新

  1. TCP协议是安全的吗?

    2023-12-24 18:22:02       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2023-12-24 18:22:02       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2023-12-24 18:22:02       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2023-12-24 18:22:02       20 阅读

热门阅读

  1. 贪心算法高频问题-区间问题

    2023-12-24 18:22:02       35 阅读
  2. vue 项目 index.html 中使用环境变量

    2023-12-24 18:22:02       42 阅读
  3. 语音信号加密算法MATLAB仿真

    2023-12-24 18:22:02       44 阅读
  4. npm使用详解(好吧好吧是粗解)

    2023-12-24 18:22:02       40 阅读
  5. 机器学习之实验过程02

    2023-12-24 18:22:02       30 阅读
  6. Starknet 命令行工具之Starkli | 使用Starkli部署合约

    2023-12-24 18:22:02       33 阅读
  7. 前端八股文(js篇 )

    2023-12-24 18:22:02       39 阅读
  8. 自动编码器图像去噪 Python

    2023-12-24 18:22:02       36 阅读
  9. 文件包含

    2023-12-24 18:22:02       30 阅读