ndarray转换为tensor的函数

在PyTorch中，可以使用torch.from_numpy()函数将numpy数组转换为tensor。例如：

import numpy as np
import torch

# 创建一个numpy数组
arr = np.array([1, 2, 3, 4, 5])

# 将numpy数组转换为tensor
tensor = torch.from_numpy(arr)

print(tensor)

输出：

tensor([1, 2, 3, 4, 5])

需要注意的是，该函数会返回一个新的tensor，但是它与原始的numpy数组共享内存，因此在修改tensor时，原始的numpy数组也会被修改。如果需要避免这种情况，可以使用tensor.clone()函数创建一个新的tensor。