Pytorch常用函数功能使用（一）-蒲公英云

Pytorch常用函数功能使用（一）

忘是亡心i 2022-03-15 14:55 335阅读 0赞

1. view

import torch
number_1 = torch.randn(2, 3)
print(number_1)
print(number_1.shape)
print(number_1.view(1, -1))
print(number_1.view(3, -1))

输出：

tensor([[ 1.0506, -0.5875, -1.2477],
        [ 0.0635,  0.8997,  0.1551]])
torch.Size([2, 3])
tensor([[ 1.0506, -0.5875, -1.2477,  0.0635,  0.8997,  0.1551]])
tensor([[ 1.0506, -0.5875],
        [-1.2477,  0.0635],
        [ 0.8997,  0.1551]])

View(a,b)中第一个参数a代表目标张量的行数，b代表列数。为了简便起见，也可以只指定第一个参数a，b这个参数设置成-1，函数会自动计算对应的列数。

2. squeeze

number_2 = torch.randn(2, 1)
print(number_2)
print(torch.squeeze(number_2))
print(torch.squeeze(number_2, 0))
print(torch.squeeze(number_2, 1))

输出：

tensor([[ 0.5856],
        [-1.7095]])
tensor([ 0.5856, -1.7095])
tensor([[ 0.5856],
        [-1.7095]])
tensor([ 0.5856, -1.7095])

Squeeze的功能是进行维度缩减（维度为1的删除）。Squeeze(a,b)中第一个参数a代表传入的张量，b代表要缩减的维数。如果第二个参数没有指定，则默认删除所有维度为1的维度

number_3 = torch.randn(1, 2)
print(number_3)
print(torch.squeeze(number_3))
print(torch.squeeze(number_3, 0))
print(torch.squeeze(number_3, 1))

输出：

tensor([[ 0.1555, -0.4286]])
tensor([ 0.1555, -0.4286])
tensor([ 0.1555, -0.4286])
tensor([[ 0.1555, -0.4286]])

3. unsqueeze

number_4 = torch.randn(3, 2)
print(number_4)
print(torch.unsqueeze(number_4, 0))
print(torch.unsqueeze(number_4, 1))

输出：

tensor([[ 0.0358, -0.2769],
        [-0.3257,  0.1895],
        [ 1.9278, -0.9444]])
tensor([[[ 0.0358, -0.2769],
         [-0.3257,  0.1895],
         [ 1.9278, -0.9444]]])
tensor([[[ 0.0358, -0.2769]],
        [[-0.3257,  0.1895]],
        [[ 1.9278, -0.9444]]])

Unsqueeze的功能与squeeze相反，可以增加张量的维度。Unqueeze(a,b)中第一个参数a代表传入的张量，b代表要增加维度的维数。

4. max

number_5 = torch.randn(2, 3)
print(number_5)
print(torch.max(number_5, 0))
print(torch.max(number_5, 1))

输出：

tensor([[-0.4916,  1.3999,  1.0527],
        [ 1.0194, -2.4695, -0.2378]])
(tensor([1.0194, 1.3999, 1.0527]), tensor([1, 0, 0]))
(tensor([1.3999, 1.0194]), tensor([1, 0]))