本篇主要讲解如何创建tensor。
0.总结
- 详细代码见我Github
- 欢迎留言提问交流
- 文章来源:CSDN_LawsonAbs
1.将list传入到tensor()中,然后得到一个tensor
- 01.指定tensor 的类型,否则就是由传入的数据类型转换得到
import torch as t
a = [1,2,3,4]
a = t.tensor(a) # 转成 tensor
print(a)
print(a.dtype) # 输出tensor的类型
b = t.tensor([1,2,],dtype=t.long)
print(b)
print(b.dtype)
2. 使用函数
- 使用
randn()
函数
- 使用
randint()
函数