Tensor的常用属性和方法
属性
requires_grad: 查看该tensor是否需要求导
grad_fn:查看该tensor经历了什么样的操作
方法
requires_grad_(): 将tensor变为可求导
cuda(): 将tensor迁移到GPU显存中
查看tensor的属性和方法
代码如下:
import torch
ten1 = torch.FloatTensor(2)
func_attrs = dir(ten1)
tmp = []
for idx, func_attr in enumerate(func_attrs):
tmp.append(func_attr)
if (idx+1)%15 == 0:
print(tmp)
tmp = []
print("ten1的类型", type(ten1))
print("func_attrs的数量:", len(func_attrs))
运行结果如下: