张量

PyTorch张量的实用函数。

函数

torch_to	尝试递归地将数据移动到指定的args/kwargs。
torch_detach	尝试递归地从计算图中分离数据。
torch_to_numpy	将torch张量转换为numpy数组。
numpy_to_torch	将numpy数组转换为torch张量。

numpy_to_torch(np_outputs)

将numpy数组转换为torch张量。

Parameters:

np_outputs (List[ndarray]) –

Return type:

列表[张量]

torch_detach(data)

尝试递归地从计算图中分离数据。

torch_to(data, *args, **kwargs)

尝试递归地将数据移动到指定的args/kwargs。

torch_to_numpy(inputs)

将torch张量转换为numpy数组。

Parameters:

输入 (列表[张量]) –

Return type:

列表[ndarray]