PyTorch基本操作
PyTorch基本操作
1. torch.allclose是一个PyTorch函数,用于检查两个张量是否在某个容忍度范围内近似相等
torch.allclose是一个PyTorch函数,用于检查两个张量是否在某个容忍度范围内近似相等
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
input (Tensor) – 第一个输入张量
other (Tensor) – 第二个输入张量
rtol (float) – 相对容忍度
atol (float) – 绝对容忍度
equal_nan (bool) – 如果为True,将视NANs在两个张量中为相等,即使它们处于不同的位置。
返回:
(bool) - 如果两个张量在容忍度范围内是近似的,则返回True,否则返回False。
两个张量a和b被视为近似相等,如果以下条件对所有元素都成立:
|a - b| <= atol + rtol * |b|
import torch
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([1.0, 2.001, 3.0])
print(torch.allclose(a, b, rtol=1e-03))
#True
print(torch.allclose(a, b, rtol=1e-05))
2. torch 查看GPU
print(torch.__version__) # pytorch版本
print(torch.version.cuda) # cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.to(device)
https://www.cnblogs.com/zhongzhaoxie/p/16692999.html
https://www.cnblogs.com/shizyxya/p/zhangyuxue.html
https://blog.csdn.net/pipisorry/article/details/108989515
https://blog.csdn.net/hhhhxxn/article/details/110119567