Pytorch numpy 与 tensor 互相转换
Pytorch中numpy 与 tensor互相转换tensor 转 numpy1.1可以使用torch.tensor()直接将numpy转换为tensor,但是此时数据会被复制,不会与原来的数据共享内存。1.2可以使用torch.from_numpy()将numpy转换为tensor,会与原来的数据共享内存。numpy 转 tensor2.1可以使用numpy.array()直接将tensor转
·
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
更多推荐
所有评论(0)