pytorch 中 antorcha.isnan () 和 antorcha.isfinite ()

1.torch.isfinite ()

import torch
num = torch.tensor(1)  # 数字1
res = torch.isfinite(num)
print(res)
'''
输出：
tensor(True)
'''

Este número debe ser tensor

import torch
num = torch.tensor(float('inf')) # 正无穷大
res = torch.isfinite(num)
print(res)
'''
输出：
tensor(False)
'''

import torch
num = torch.tensor(float('-inf')) # 负无穷大
res = torch.isfinite(num)
print(res)
'''
输出：
tensor(False)
'''

import torch
num = torch.tensor(float('nan')) # 空
res = torch.isfinite(num)
print(res)
'''
输出：
tensor(False)
'''

2.torch.isnan ()

import torch
res=torch.isnan(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')]))
print(res)
'''
输出：
tensor([False, False, False, False,  True])
'''

Se puede ver que torch.isnan () es una función que se usa para determinar si el tensor de entrada está vacío. Cuando la entrada está vacía, devuelve True.

 

 

Realice programación en Matlab, Python y C ++, aprendizaje automático, implementación y orientación de la teoría de la visión por computadora, tanto de pregrado como de maestría, comercio de pescado salado, respuestas profesionales, por favor, comuníquese con el número de QQ 757160542, si usted es el indicado