torch.is_tensor() 和 isinstance()1 torch.is_tensor()示例2 isinstance()

100 阅读 0 评论 66 点赞

我是靠谱客的博主勤劳蜜蜂，最近开发中收集的这篇文章主要介绍torch.is_tensor() 和 isinstance()1 torch.is_tensor()示例2 isinstance()，觉得挺不错的，现在分享给大家，希望可以做个参考。

1 torch.is_tensor()

用来判断对象是不是 tensor, 是tensor的话返回true，否则返回false

torch.is_tensor(obj)

import torch

a = torch.tensor([[1, 1], [2, 2]])
torch.is_tensor(a)

>>True

isinstance(obj, type)

这个方法可以检查任何类型，如果检查出obj是type类型返回true，否则返回false

import torch

a = torch.tensor([[1, 1], [2, 2]])
isinstance(a, Tensor)

>>name 'Tensor' is not defined

注意 isinstance() 里面的 tensor 类型不是 Tensor, 而是 torch.Tensor

import torch

a = torch.tensor([[1, 1], [2, 2]])
print(type(a))
isinstance(a, torch.Tensor)

>><class 'torch.Tensor'>
>>True

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。