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:查看张量的形状
tenso