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.]]])
相关推荐
智算菩萨8 小时前
Anthropic Claude 4.5:AI分层编排的革命,成本、速度与能力的新平衡
前端·人工智能
小Pawn爷8 小时前
12. 智能与风险并存:金融AI的成本,合规与伦理平衡术
人工智能·金融·llm·合规
●VON8 小时前
AI 保险机制:为智能时代的不确定性兜底
人工智能·学习·安全·制造·von
开发者导航8 小时前
【开发者导航】一键解决AI生成内容格式复制难题的剪贴板工具:PasteMD
人工智能
代码游侠8 小时前
学习笔记——HC-SR04 超声波测距传感器
开发语言·笔记·嵌入式硬件·学习
bu_shuo9 小时前
将AI生成的数学公式正确复制到word中
人工智能·chatgpt·word·latex
AI科技星9 小时前
光速飞行器动力学方程的第一性原理推导、验证与范式革命
数据结构·人工智能·线性代数·算法·机器学习·概率论
摘星编程9 小时前
RAG的下一站:检索增强生成如何重塑企业知识中枢?
android·人工智能
Aaron_9459 小时前
BitNet:1-bit大语言模型的高效推理框架详解
人工智能·语言模型·自然语言处理
wenzhangli79 小时前
「1+3 架构驱动」OoderAI 企业级解决方案:破解 AI 落地三大痛点,实现能力可控、交互智能与代码一致
人工智能