1 Star 0 Fork 0

K-on/机器学习

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
tensor属性.py 467 Bytes
一键复制 编辑 原始数据 按行查看 历史
K-on 提交于 2020-10-09 10:49 . tensor基础
import torch
# 创建一个tensor对象, 类似于 numpy
a = torch.tensor([1, 2, 3], dtype=int)
print(a)
# 查看对象的类型 dtype
print(a.dtype)
b = torch.tensor([4, 5, 6], dtype=float)
print(b)
# 调用ndimension()方法来查看 tensor 的维度
tensor = torch.tensor([[1, 2, 3],
[3, 4, 5]])
print(tensor.ndimension())
# 调用 shape 或者 size() 来查看数据的形状(就是行和列)
print(tensor.shape)
# print(tensor.size())
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/HuangJiaLuo/machine-learning.git
[email protected]:HuangJiaLuo/machine-learning.git
HuangJiaLuo
machine-learning
机器学习
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385