zoukankan      html  css  js  c++  java
  • Pytorch tensor的复制函数torch.repeat_interleave()

    1. repeat_interleave(self: Tensor, repeats: _int, dim: Optional[_int]=None)

    参数说明:

    self: 传入的数据为tensor

    repeats: 复制的份数

    dim: 要复制的维度,可设定为0/1/2.....

    2. 例子

    2.1 Code

    此处定义了一个4维tensor,要对第2个维度复制,由原来的1变为3,即将设定dim=1。

     1 import torch
     2 
     3 
     4 def function():
     5     data1 = torch.rand([2, 1, 3, 3])
     6     print("data1_shape: ", data1.shape)
     7     print("data1: ", data1)
     8 
     9     data2 = torch.repeat_interleave(data1, repeats=3, dim=1)
    10     print("data2_shape: ", data2.shape)
    11     print("data2: ", data2)
    12 
    13 
    14 if __name__ == '__main__':
    15     function()
    View Code

    2.2 输出显示

    即可看到输入tensor形状为[2, 1, 3, 3],经过repeat后,tensor变为[2, 3, 3, 3],并在第二维度上保持相同的数据。

     

  • 相关阅读:
    时间选择框(可用于Form)
    点击复制指定内容
    ajax中多个模板之间套用ajax
    Java学习路径
    Windows平台安装Python
    Python语法-第2关
    Python语法-第1关
    Python语法-第0关
    图像识别
    wx:for用法
  • 原文地址:https://www.cnblogs.com/haifwu/p/12814760.html
Copyright © 2011-2022 走看看