【属性】pytorch-tensor对象的属性

本站原创文章，转载请说明来自《老饼讲解-深度学习》www.bbbdata.com

在日常使用pytorch的tensor对象时，往往需要查看tensor对象的一些大小、数据类型等等基本属性

本文讲解如何查看pytorch中tensor对象的属性，并展示相关示例

一. 查看tensor对象的属性  

本节讲解如何查看tensor对象的常用属性

01.如何查看tensor对象的常用属性

tensor对象的常用属性例如大小、数据类型、是否在GPU等等属性的查看方法如下：

import torch
x = torch.tensor( [[1, 2],[3, 4]])   # 生成一个tensor数据x
#  常用基本属性
print('tensor的大小:',x.shape)                           
print('tensor的数据类型:',x.dtype)                           
print('tensor的所在设备:',x.device) 

# tensor的数据与梯度
print('tensor的数据:',x.data) 
print('tensor是否有梯度:',x.requires_grad) 
print('tensor的梯度:',x.grad) 

# 其它
print('tensor的视图:',x.view) 
print('tensor的存储器:',x.storage)                          

运行结果如下：

tensor的大小: torch.Size([2, 2])
tensor的数据类型: torch.int64
tensor的所在设备: cpu
tensor的数据: tensor([[1, 2],
                     [3, 4]])
tensor是否有梯度: False
tensor的梯度: None
tensor的视图: <built-in method view of Tensor object at 0x00000122E9A74B80>
tensor的存储器: <bound method Tensor.storage of tensor([[1, 2], [3, 4]])>

好了，以上就是在pytorch中如何查看tensor对象的属性的内容了~

 End