pytorch笔记:ReplicationPad1d

python 复制代码
torch.nn.ReplicationPad1d(padding)
  • 在 PyTorch 中,ReplicationPad1d 是一种用于一维数据的填充层
  • 该层通过复制序列的边缘值来增加数据的长度,这在卷积神经网络中常用于保持数据尺寸
  • 主要参数

|---------|-----------------------------------------------------------------------------------------------------|
| padding | 可以是一个整数或一个元组。 * 如果是一个整数,它表示在序列的每一端都填充相同数量的值。 * 如果是一个元组 (pad_left, pad_right),则分别在序列的左边和右边填充指定数量的值 |

  • 工作原理:

    • 假设有一个序列 [a, b, c, d],并且设置 padding=(2, 3),那么填充后的序列将是 [a, a, a, b, c, d, d, d, d]

    • 这里,左边的 a 被复制了两次,右边的 d 被复制了三次。

  • 举例:

python 复制代码
import torch
import torch.nn as nn

# 创建一维数据
data = torch.tensor([1, 2, 3, 4], dtype=torch.float32).unsqueeze(0).unsqueeze(0)  # shape: [1, 1, 4]

data,data.shape
#(tensor([[[1., 2., 3., 4.]]]), torch.Size([1, 1, 4]))




pad = nn.ReplicationPad1d((2, 3))

# 应用填充
padded_data = pad(data)

print(padded_data) 
#tensor([[[1., 1., 1., 2., 3., 4., 4., 4., 4.]]])
相关推荐
锋行天下4 小时前
公司内网部署大模型的探索之路
前端·人工智能·后端
智者知已应修善业5 小时前
【输入两个数字,判断两数相乘是否等于各自逆序数相乘】2023-10-24
c语言·c++·经验分享·笔记·算法·1024程序员节
集3045 小时前
C++多线程学习笔记
c++·笔记·学习
YJlio5 小时前
Active Directory 工具学习笔记(10.0):AdExplorer / AdInsight / AdRestore 导读与场景地图
网络·笔记·学习
背心2块钱包邮6 小时前
第7节——积分技巧(Integration Techniques)-代换积分法
人工智能·python·深度学习·matplotlib
广东数字化转型6 小时前
工作备注笔记
笔记
无心水6 小时前
【分布式利器:大厂技术】4、字节跳动高性能架构:Kitex+Hertz+BytePS,实时流与AI的极致优化
人工智能·分布式·架构·kitex·分布式利器·字节跳动分布式·byteps
阿正的梦工坊6 小时前
DreamGym:通过经验合成实现代理学习的可扩展化
人工智能·算法·大模型·llm
超高校级的作者6 小时前
博客摘录「 CentOS7 Fail2ban安装使用」2024年4月15日
笔记
湘-枫叶情缘6 小时前
人脑生物芯片作为“数字修炼世界”终极载体的技术前景、伦理挑战与实现路径
人工智能