列表转数组:np.array(list)

数组转列表:np.array.tolist()

列表转张量:torch.Tensor(list)

张量转列表:先转numpy,后转list,tensor.numpy().tolist()

张量转数组:tensor.numpy()

注:在 gpu 上的张量不能直接转为numpy,需要使用cpu(),tensor.cpu().numpy()

数组转张量:torch.from_numpy(ndarray),tf.conver_to_tensor(ndarray)

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐