pytorch-pytorch之LSTM

1. nn.LSTM

初始化函数输入参数与RNN相同,分别是input_size,hidden_size和num_layer

在这里插入图片描述
foward函数也与RNN类似,只不过返回值除了out外,ht变为(ht,ct)
在这里插入图片描述
代码见下图:
在这里插入图片描述

2. nn.LSTMCell

初始化函数输入参数与nn.LSTM类似
在这里插入图片描述
forward函数与nn.LSTM的区别是没有out
在这里插入图片描述
单层代码如下图:
和RNN的Cell类似,Cell单元循环更新h,c
在这里插入图片描述
2层无非就是多创建了一个cell2、h2、c2,并同时循环更新h1,c1,h2,c2,多层以此类推。
在这里插入图片描述

相关推荐

  1. PyTorchPyTorch包装容器

    2024-07-14 09:20:02       36 阅读
  2. <span style='color:red;'>Pytorch</span>

    Pytorch

    2024-07-14 09:20:02      50 阅读
  3. PyTorch

    2024-07-14 09:20:02       49 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-14 09:20:02       101 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-14 09:20:02       109 阅读
  3. 在Django里面运行非项目文件

    2024-07-14 09:20:02       87 阅读
  4. Python语言-面向对象

    2024-07-14 09:20:02       96 阅读

热门阅读

  1. 大语言模型LLM

    2024-07-14 09:20:02       22 阅读
  2. 大模型时代,还需要跨端framework吗?

    2024-07-14 09:20:02       30 阅读
  3. 搭建docker私有仓库

    2024-07-14 09:20:02       30 阅读
  4. uvm中使用clone时,为什么要使用$cast

    2024-07-14 09:20:02       26 阅读
  5. Linux命令怎么背?

    2024-07-14 09:20:02       25 阅读
  6. OpenCV npy和txt文件的区别与使用

    2024-07-14 09:20:02       29 阅读