a=torch.randn((3,2))
tensor转换为numpy:numpy_a=a.numpy()
numpy转换为tensor:torch_a=torch.from_numpy(numpy_a)
需要特别注意的是,tensor和numpy对象共享内存,所以他们之间转换很快,而且几乎不消耗什么资源,但这也意味着,如果其中一个变了,另外一个也随之改变,可以简单地理解为始终在一个对象上进行转换操作,不管类型变成何种类型,但是值始终跟随着变化。
a=torch.randn((3,2))
tensor转换为numpy:numpy_a=a.numpy()
numpy转换为tensor:torch_a=torch.from_numpy(numpy_a)
需要特别注意的是,tensor和numpy对象共享内存,所以他们之间转换很快,而且几乎不消耗什么资源,但这也意味着,如果其中一个变了,另外一个也随之改变,可以简单地理解为始终在一个对象上进行转换操作,不管类型变成何种类型,但是值始终跟随着变化。