to_tensor接口

接口功能

将数据转换为torch.tensor。

接口实现

def to_tensor(self, data):
        return torch.tensor(data, device=self.device)

参数说明

参数名称

是否必选

类型

默认值

描述

data

ndarry

-

待转为torch.tensor的数据。