pytorch中的torch.from_numpy()

https://blog.csdn.net/qq_26020233/article/details/89007762
Tensor中并不储存数据,而是存储有对应numpy数组的数据指针的拷贝

import numpy as np
import torch
a = np.array([1,2,3])
b = torch.from_numpy(a)
# b = torch.tensor(a)
print("b 变化前:",b)

b += 1

print("a 变化后:",a)
print("b 变化后:",b)
print("a的地址:",id(a))
print("b的地址:",id(b))

'''
b 变化前: tensor([1, 2, 3], dtype=torch.int32)
a 变化后: [2 3 4]
b 变化后: tensor([2, 3, 4], dtype=torch.int32)
a的地址: 1721197547312
b的地址: 1720812881144

'''
  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值