Pytorch中numpy 与 tensor互相转换

1.numpy 转 tensor


    可以使用torch.tensor()直接将numpy转换为tensor,但是此时数据会被复制,不会与原来的数据      共享内存。

    可以使用torch.from_numpy()将numpy转换为tensor,会与原来的数据共享内存。

2.tensor 转 numpy


    可以使用numpy.array()直接将tensor转换为numpy。

    也可以使用torch.numpy()将tensor转换为numpy。

参考:https://blog.csdn.net/shey666/article/details/85337212

Logo

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

更多推荐