formatting Namespace Reference

Classes
class	CollectKeys
class	ImageToTensor
class	ToTensor

Functions
	to_tensor (data)

Function Documentation

to_tensor()

formatting.to_tensor

(

data

)

데이터를 텐서로 변환하는 기능
args:
    data : 다양한 타입의 데이터
return: Tensor로 변환된 데이터

Definition at line 6 of file formatting.py.

def to_tensor(data):
    """
    데이터를 텐서로 변환하는 기능
    args:
        data : 다양한 타입의 데이터
    return: Tensor로 변환된 데이터
    """
    if isinstance(data, torch.Tensor):
        return data
    elif isinstance(data, np.ndarray):
        return torch.from_numpy(data)
    elif isinstance(data, Sequence) and not isinstance(data, str):
        return torch.tensor(data)
    elif isinstance(data, int):
        return torch.LongTensor([data])
    elif isinstance(data, float):
        return torch.FloatTensor([data])
    elif isinstance(data, dict):
        for key in data.keys():
            data[key] = to_tensor(data[key])
    else:
        raise TypeError(f'to_tensor : type {type(data)} cannot be converted to tensor.')