本篇为自己在学习时整理出的Tensor笔记。
目录
产生张量:
在生成张量时,按照需求的不同,有以下方法:
可在()内加入dtype=___,去控制Datatype,有:torch.float32 / torch.int64 / torch.float16 / torch.uint8 / torch.bool…等等。
0张量 | torch.zeros(row,col) →可在里面输入0得空tensor |
1张量 | torch.ones(row,col) |
单位张量 | torch.eye(rank) |
均匀随机 | torch.rand(row,col) |
填 |