pytorch 張量 張量的資料型別

2021-10-10 17:29:59 字數 991 閱讀 7252

張量定義:

import torch

torch.tensor(

[1.2

,3.4])

.dtype #獲取張量的資料型別,其中torch.tensor()函式生成乙個張量

# torch.float32

torch.set_default_tensor_type(torch.doubletensor)

#設定張量的預設資料型別

torch.tensor(

[1.2

,3.4])

.dtype

#torch.float64

在torch中還有其他型別的資料,將浮點型轉化為其他資料型別的方法如下:

a = torch.tensor(

[1.2

,3.4])

print

("a.dtype:"

,a.dtype)

print

("a.long()方法"

,a.long()

.dtype)

print

("a.int()方法"

,a.int()

.dtype)

print

("a.float()方法"

,a.float()

.dtype)

##a.dtype:torch.float64

#a.long()方法:torch.int64

#a.int()方法:torch.int32

#a.float()方法:torch.float32

一系列轉化之後還可以通過torch.set_default_tendor_type()來設定預設資料型別,也可以使用torch.get_default_dtype()函式獲取預設的資料型別,**如下:

torch.get_default_dtype(

)#torch.float32

pytorch 張量 張量的生成

張量的生成 import torch import numpy as np 使用tensor.tensor 函式構造張量 a torch.tensor 1.0,1.0 2.2 print a 獲取張量的維度 print 張量的維度 a.shape 獲取張量的形狀大小 print 張量的大小 a.si...

pytorch張量追蹤

torch.tensor 是這個包的核心類。如果設定它的屬性 requires grad 為 true,那麼它將會追蹤對於該張量的所有操作。當完成計算後可以通過呼叫 backward 來自動計算所有的梯度。這個張量的所有梯度將會自動累加到.grad屬性.要阻止乙個張量被跟蹤歷史,可以呼叫 detac...

pytorch 張量重構view

view在pytorch中是用來改變張量的shape的,簡單又好用。pytorch中view的用法通常是直接在張量名後用.view呼叫,然後放入自己想要的shape。如 tensor name.view shape example 1.直接用法 x torch.randn 4,4 x.size to...