创建pytorch张量

1、numpy->tensor

注:

tensor()参数为具体的数据, Tensor() 和 FloatTensor()参数可以为数据也可以为维度信息,但是一般不用他来放数据
a  = torch.Tensor(2, 3)
print(a)

a = torch.FloatTensor(2, 3)
print(a)
a = np.array([2., 3])

b = torch.from_numpy(a)
print(b)
###################################
a = np.ones([2, 3])
b = torch.from_numpy(a)
print(b)

2、list->tensor

a = torch.tensor([[1., 2., 3.],
                 [4., 5., 6.]])
print(a)

3、创建空的tensor(这种方法之后一定要进行初始化,否则由于有的值为INF 或 -INF,可能出现一些玄学错误)

a = torch.Tensor(2, 3)
print(a)

b = torch.FloatTensor(2, 3)
print(b)

c = torch.IntTensor(2, 3)
print(c)

 notes:修改默认tensor类型:set_default_tensor_type


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值