PyTorch:torch.Tensor.repeat()、expand()

目录

1、torch.Tensor.repeat()

2、torch.Tensor.expand()


1、torch.Tensor.repeat()

函数定义:

repeat(*sizes) → Tensor

作用:

在指定的维度上重复这个张量,即把这个维度的张量复制*sizes次。同时可以通过复制的形式扩展维度的数量。

注意:torch.Tensor.repeat方法与numpy.tile方法作用相似,而不是numpy.repeat!torch中与numpy.repeat类似的方法是torch.repeat_interleave!

区别:与expand的不同之处在于,repeat函数传入的参数直接就是对应维度要扩充的倍数,而不是最后的shape。

举例分析:

例1——对应(已存在的)维度的拓展。

import torch

a = torch.tensor([[1], [2], [3]])  # 3 * 1
b = a.repeat(3, 2)  # torch.linspace(0, 10, 5)
print('a:\n', a)
print('shape of a', a.size())  # 原始shape = (3,1)
print('b:\n', b)
print('shape of b', b.size())  # 新的shape = (3*3,1*2),新增加的数据通过复制得到

'''   运行结果   '''
a:
 tensor([[1],
        [2],
        [3]])
shape of a torch.Size([3, 1])  注: 原始shape = (3,1)
b:
 tensor([[1, 1],
        [2, 2],
        [3, 3],
        [1, 1],
        [2, 2],
        [3, 3],
        [1, 1],
        [2, 2],
        [3, 3]])
shape of b torch.Size([9, 2])  注: 新的shape = (3*3,1*2),新增加的数据通过复制得到

例2——带有(原始不存在的)维度数量拓展的用法:

import torch
a = torch.tensor([[1, 2], [3, 4], [5, 6]])  # 3 * 2
b = a.repeat(3, 2, 1)   # 在原始tensor的0维前拓展一个维度,并把原始tensor的第1维扩张2倍,都是通过复制来完成的
print('a:\n', a)
print('shape of a', a.size())  # 原始维度为 (3,2)
print('b:\n', b)
print('shape of b', b.size())  # 新的维度为 (3,2*2,2*1)=(3,4,2)

'''   运行结果   '''
a:
 tensor([[1, 2],
         [3, 4],
         [5, 6]])
shape of a torch.Size([3, 2])   注:原始维度为 (3,2)
b:
 tensor([[[1, 2],
          [3, 4],
          [5, 6],
          [1, 2],
          [3, 4],
          [5, 6]],
 
         [[1, 2],
          [3, 4],
          [5, 6],
          [1, 2],
          [3, 4],
          [5, 6]],

         [[1, 2],
          [3, 4],
          [5, 6],
          [1, 2],
          [3, 4],
          [5, 6]]])
shape of b torch.Size([3, 6, 2])   注:新的维度为 (3,2*2,2*1)=(3,4,2)

2、torch.Tensor.expand()

函数定义:

expand(*sizes) → Tensor

作用:不仅可以对tensor指定的(已存在的)维度进行扩大(复制型扩大),扩大后的shape为*(size)。而且还有类似于unsqueeze的维度扩充功能,新增加的维度将会加在前面。

区别:与repeat不同之处在于,expand传入的参数直接就是将tensor扩大后的shape。

举例说明

a = torch.ones(3, 1)   # 创建3*1的全为1的tensor
b = a.expand(3, 2)     # 对a的维度1进行扩充
c = a.expand(2, 3, 2)  # 对a的维度1进行扩充,并在维度0前加一个维度 
print('a:', a)
print('shape of a:', a.shape)
print('b:', b)
print('shape of b:', b.shape)
print('c:', c)
print('shape of c:', c.shape)

'''   运行结果   '''
a: tensor([[1.],
           [1.],
           [1.]])
shape of a: torch.Size([3, 1])

b: tensor([[1., 1.],
           [1., 1.],
           [1., 1.]])
shape of b: torch.Size([3, 2])

c: tensor([[[1., 1.],
            [1., 1.],
            [1., 1.]],
           [[1., 1.],
            [1., 1.],
            [1., 1.]]])
shape of c: torch.Size([2, 3, 2])

你可能感兴趣的:(Pytorch,pytorch)