欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

如何判断一个对象是否为Tensor?

程序员文章站 2022-06-10 21:00:31
...

如何判断一个对象是否为Tensor?

import torch
import numpy as np
obj = np.arange(1,10)  # arange() 主要是用于生成数组
# [1,2,3,4,5,6,7,8,9]
print(torch.is_tensor(obj))
obj1 = torch.Tensor(10)    # 生成单精度浮点类型的张量
# tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
print(torch.is_tensor(obj1))

False
True