网站的色彩拉人头最暴利的app

• 从tensor到ndarray：.detach() 方法用于创建一个新的张量，新张量与原始张量共享数据内存，但是不会被计算图追踪。这意味着在新张量上的操作不会影响到原始张量，同时也可以避免梯度传播，适合于提取中间结果。

# 当tensor在cuda上时，转换为ndarray
ndarray = tensor.cpu().detach().numpy()

• 从ndarray到tensor：

tensor = torch.from_numpy(ndarray)