def device() ->torch.device

函数功能

返回Stream所隶属的Device。

函数原型

def device() ->torch.device

返回值说明

torch.device类型，返回Stream所隶属的Device。