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.]]])
相关推荐
IT_陈寒10 分钟前
Redis性能翻倍的5个冷门技巧:从每秒10万到20万的实战优化之路
前端·人工智能·后端
HXR_plume21 分钟前
【Web信息处理与应用课程笔记4】个性化检索(下)
笔记
鹿鸣天涯24 分钟前
DeepSeek发布新论文提出更为高效的AI开发方法
人工智能
im_AMBER29 分钟前
Leetcode 94 合并零之间的节点
数据结构·c++·笔记·学习·算法·leetcode
酌沧30 分钟前
拆解GUI Agent大模型利用PPO强化学习
人工智能·智能手机
Coder_Boy_31 分钟前
基于SpringAI的企业级智能教学考试平台模块四:优化迭代模块内容
人工智能·spring boot
我是宝库36 分钟前
SCI论文在哪些情况下可以不查AI率?
人工智能·aigc·sci论文·turnitin系统·英文查重·aigc检测·sci发表
专注前端30年40 分钟前
AI工程化_MLOps实战:模型部署+FastAPI+ONNX Runtime+TensorRT加速全解析
人工智能·fastapi
编码小哥41 分钟前
OpenCV角点检测:Harris与ShiTomasi算法
人工智能·opencv·算法
杨建允1 小时前
杨建允:AI搜索优化对工业品行业的影响
人工智能·ai