libcity.data.batch¶

class libcity.data.batch.Batch(feature_name)[源代码]¶

基类：object

append(item)[源代码]¶

append a new item into the batch

参数: item (list) – 一组输入，跟feature_name的顺序一致，feature_name即是这一组输入的名字

to_ndarray()[源代码]¶

to_tensor(device)[源代码]¶

将数据self.data转移到device上

参数: device (torch.device) – GPU/CPU设备

class libcity.data.batch.BatchPAD(feature_name, pad_item=None, pad_max_len=None)[源代码]¶

基类：libcity.data.batch.Batch

append(item)[源代码]¶

append a new item into the batch

参数: item (list) – 一组输入，跟feature_name的顺序一致，feature_name即是这一组输入的名字

get_origin_len(key)[源代码]¶

padding()[源代码]¶: 只提供对一维数组的特征进行补齐

to_tensor(device)[源代码]¶

将数据self.data转移到device上

参数: device (torch.device) – GPU/CPU设备