将n维numpy数组转换为火炬张量

Question

我有4维numpy数组（type(x)是ndarray，type(x[i])是ndarray，type(x[i][j])是ndarray和{{ 1}}是type(x[i][j][k])）。我需要将其转换为火炬张量，以将其用作数据集的一部分。我试图像这样堆叠float32：

torch.Tensor

但不确定是否能奏效。 有更简单或更方便的方法吗？

P.S。 X看起来像这样：

def convert_to_tensor(ndarr):
    for i in ndarr:
        for j in ndarr[i]:
            tensor_ndarr = torch.stack([torch.Tensor(k) for k in ndarr[i][j]])
    return tensor_ndarr