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

2021-09-11 19:51:20 字數 2488 閱讀 5209

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，函式會自動計算對應的列數。

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]])
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代表要增加維度的維數。

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]))
max的功能是返回對應維度最大的數以及對應的索引。max(a,b)中第乙個引數a代表傳入的張量，b代表要對應的維數。0代表返回每一列的最大值，1代表返回每一行的最大值。

vim功能使用

行首 0 fn 左箭頭 行尾 fn 右箭頭 文首 gg 文尾 g 某行 ng eg 4g表示游標跳到第四行 游標右移n字元 n eg 4表示游標右移4字元 刪除游標所在行 dd 刪除游標所在n行 ndd 刪除後乙個字元 x 刪除前乙個字元 x 刪除後n個字元 nx 刪除前n個字元 nx 複製游標所在...

MF CListControl 簡單功能使用

可檢視 todo 在此新增額外的初始化 ananlydata.modifystyle 0,lvs showselalways lvs ex gridlines lvs ex twoclickactivate ananlydata.setextendedstyle lvs ex flatsb 扁平風格...

pytorch常用函式

torch.cat torch.squeeze torch.unsqueeze torch.stack torch.sum torch.sum input dim,out none tensor input tensor 輸入張量 dim int 縮減的維度 out tensor,optional ...