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.]]])
相关推荐
程序员水自流9 分钟前
【AI大模型第9集】Function Calling,让AI大模型连接外部世界
java·人工智能·llm
手揽回忆怎么睡10 分钟前
Streamlit学习实战教程级,一个交互式的机器学习实验平台!
人工智能·学习·机器学习
小徐Chao努力12 分钟前
【Langchain4j-Java AI开发】06-工具与函数调用
java·人工智能·python
db_murphy20 分钟前
时事篇 | Manus收购
人工智能
攻城狮7号24 分钟前
阶跃星辰开源NextStep-1.1图像模型:告别“鬼影”与“马赛克”?
人工智能·ai图像生成·nextstep-1.1·阶跃星辰开源模型·图像模型
_codemonster29 分钟前
BERT中的padding操作
人工智能·深度学习·bert
笙枫41 分钟前
基于AI Agent框架下的能源优化调度方案和实践 | 架构设计
人工智能·能源
杭州泽沃电子科技有限公司1 小时前
面对风霜雨雪雷电:看在线监测如何为架空线路筑牢安全网
运维·人工智能·在线监测·智能监测
小真zzz1 小时前
Nano Banana Pro与Banana系产品全面解析,深度集成Nano Banana Pro的编辑能力标杆
人工智能·ai·powerpoint·ppt·nano banana pro
睡醒了叭1 小时前
图像分割-深度学习-FCN模型
人工智能·深度学习·计算机视觉