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.]]])
相关推荐
Coisinilove3 分钟前
MATLAB学习笔记——第二章
笔记·学习·matlab
美酒没故事°6 分钟前
AI中的agent、skill、mcp都是什么?
人工智能·ai
Titan20249 分钟前
C++异常学习笔记
c++·笔记·学习
后端小肥肠20 分钟前
从n8n到Claude Skills:轻松搞定小红书热门美食手账,3分钟出图,小白也能会!
人工智能·aigc·agent
之歆26 分钟前
Coze 照片知识库深度解析:当 AI 学会「看图说话」
人工智能
苡~31 分钟前
【claude skill系列 - 10】Claude_Skill全栈实战_从0到1构建个人AI助手
人工智能·ai编程·api 中转站·稳定ai编程工具
小陈phd32 分钟前
多模态大模型学习笔记(五)—— 神经网络激活函数完整指南
人工智能·笔记·神经网络·学习·自然语言处理
曦云沐35 分钟前
第四篇:LangChain 1.0 Community 生态全览:第三方集成与厂商包最佳实践
人工智能·langchain·大模型开发框架
小叮当⇔42 分钟前
电动工具品牌简介
大数据·人工智能
wdfk_prog43 分钟前
[Linux]学习笔记系列 -- [drivers]char
linux·笔记·学习