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.]]])
相关推荐
OpenMiniServer3 分钟前
石化能源文明 vs 电气化能源文明
大数据·人工智能·能源
程序员泠零澪回家种桔子3 分钟前
OpenManus开源自主规划智能体解析
人工智能·后端·算法
愚公搬代码4 分钟前
【愚公系列】《AI短视频创作一本通》006-短视频成败之根本(确定短视频选题的方法和技巧)
人工智能
工具人呵呵4 分钟前
[嵌入式AI从0开始到入土]21_基于昇腾310P RC模式的Pi0模型部署实践
人工智能
_张一凡5 分钟前
【AIGC面试面经第七期】旋转位置编码RoPE:从 2D 到 nD 的完美扩展之旅
人工智能·面试·aigc
Coding_Doggy6 分钟前
服务器迁移 | psql大批量数据以及navicat重新连接、conda 环境迁移
linux·数据库·人工智能
猿小猴子9 分钟前
主流 AI IDE 之一的 OpenCode 介绍
ide·人工智能·ai·opencode
一个处女座的程序猿9 分钟前
AGI之Multi-Agent之Moltbook:《The Anatomy of the Moltbook Social Graph》翻译与解读
人工智能·microsoft·multi-agent·moltbook
hans汉斯9 分钟前
国产生成式人工智能解决物理问题能力研究——以“智谱AI”、“讯飞星火认知大模型”、“天工”、“360智脑”、“文心一言”为例
大数据·人工智能·算法·aigc·文心一言·汉斯出版社·天工
这是个栗子10 分钟前
AI辅助编程(一) - ChatGPT
前端·vue.js·人工智能·chatgpt