AI学习目录汇总

1、torch.arange

返回一维张量（一维数组），官网说明，常见的三种用法如下

输入：torch.arange(5)
输出：tensor([0, 1, 2, 3, 4])
输入：torch.arange(5, 16)
输出：tensor([ 5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
输入：torch.arange(1, 25, 2) 
输出：tensor([ 1,  3,  5,  7,  9, 11, 13, 15, 17, 19, 21, 23])

2、tensor.shape、tensor.reshape

tensor.shape：查看张量的形状
tensor.reshape：返回改变形状后的张量，原张量不变

输入：x=torch.arange(12)
输入：x
输出：tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
输入：x.shape
输出：torch.Size([12])