如何检查对象是否是 PyTorch 张量?
要检查对象是否为张量,我们可以使用该方法。如果输入是张量,则返回True;否则为假。torch.is_tensor()
语法
torch.is_tensor(input)
参数
input–要检查的对象,如果它是张量或不是。
输出结果
如果输入是张量,则返回True;否则为假。
脚步
导入所需的库。所需的库是torch。
定义张量或其他对象。
检查创建的对象是否是张量或不使用.torch.is_tensor(input)
显示结果。
示例1
# import the required library import torch # create an object x x = torch.rand(4) print(x) # check if the above created object is a tensor print(torch.is_tensor(x))输出结果
tensor([0.9270, 0.2194, 0.2078, 0.5716]) True
示例2
# import the required library
import torch
# define an object x
x = 4
# check if the above created object is a tensor
if torch.is_tensor(x):
print ("输入对象是一个张量。")
else:
print ("输入对象不是张量。")输出结果输入对象不是张量。
在示例2中,返回False,因此输入对象不是张量。torch.is_tensor(x)
热门推荐
10 圣诞祝福语简短小学
11 祖国七十华诞简短祝福语
12 老师送的祝福语简短
13 生日祝福语大全女生简短
14 祝女性生日祝福语简短
15 牛年女神节祝福语简短
16 情人表白祝福语简短大气
17 老公开业祝福语简短
18 官宣新年祝福语简短