to_tensor接口

接口功能

将数据转换为torch.tensor。

接口实现

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

参数说明

参数名称

是否必选

类型

默认值

描述

安全声明

data

ndarray

-

待转为torch.tensor的数据。

推理强依赖数据的合法性,需由用户保证。