Python PyTorch is_storage()用法及代碼示例

PyTorch torch.is_storage()如果obj是PyTorch存儲對象，則方法返回True。

用法： torch.is_storage(object)

參數

object:這是要測試的輸入張量。

返回：它返回True或False。

讓我們借助幾個示例來了解這個概念：
範例1：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatStorage(6) 
print(a) 
  
# Applying the is_storage function and  
# storing the result in 'out' 
out = torch.is_storage(a) 
print(out)

輸出：

-30587.265625
 4.555761437366413e-41
 3.906847023468105e-37
 0.0
 4.484155085839415e-44
 0.0
[torch.FloatStorage of size 6]
True

範例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.randn(4, 6) 
print(a) 
  
# Applying the is_storage function and  
# storing the result in 'out' 
out = torch.is_storage(a) 
print(out)

輸出：

0.4837 -0.3108  0.5945 -0.5967  1.0831  0.0967
-0.0174  0.4937  0.9900 -0.0532 -0.4830 -0.4711
 1.3607 -1.0815  0.4421 -0.2668 -0.5778 -0.0482
-2.0869 -0.8338  1.2260 -0.8849  0.7303 -1.0050
[torch.FloatTensor of size 4x6]
False

相關用法

注：本文由純淨天空篩選整理自PranchalKatiyar大神的英文原創作品 Python – PyTorch is_storage() method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。