PyTorch torch.is_tensor()
如果传递的对象是PyTorch张量,则方法返回True。
用法:
代码语言:javascript复制torch.is_tensor(object)
参数
- object:这是要测试的输入张量。
返回:它返回True或False。
让我们借助几个示例来了解这个概念: 范例1:
代码语言:javascript复制# Importing the PyTorch library
import torch
# A constant tensor of size n
a = torch.FloatTensor([1, 4, 6, 9])
print(a)
# Applying the is_tensor function and
# storing the result in 'out'
out = torch.is_tensor(a)
print(out)
输出:
代码语言:javascript复制1
4
6
9
[torch.FloatTensor of size 4]
True
范例2:
代码语言:javascript复制# Importing the PyTorch library
import torch
# A constant tensor of size n
a = torch.randn(4, 6)
print(a)
# Applying the is_tensor function and
# storing the result in 'out'
out = torch.is_tensor(a)
print(out)
输出:
代码语言:javascript复制 0.7491 -1.5987 -0.9733 0.0436 -0.3093 2.0007
0.5679 -0.0092 -0.2573 0.9173 2.9849 -2.0159
-1.9215 -0.9131 -0.8244 0.4160 -0.3855 0.7033
1.7367 -1.1454 -1.4369 -0.9856 -0.9076 0.6267
[torch.FloatTensor of size 4x6]
True