pytorch
likyoo
You can visit https://github.com/likyoo for more information.
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch 张量的操作
Tensor Operation1. 张量拼接与切分torch.cat():将张量的维度dim进行拼接,不会扩张张量的维度 如dim=0,则两个向量将在第0维进行拼接:(3,4)concat(3,4)-->(6,4)torch.stack():在新创建的维度dim上进行拼接 如dim=0,则(3,...原创 2020-01-14 13:53:37 · 833 阅读 · 0 评论 -
pytorch tensor创建
create tensor1. torch.tensor():从data创建tensordata可以是list,numpy;dtype默认与data一致。2. torch.from_numpy(ndarray):从numpy创建tensor这里创建的tensor与原ndarray共享内存,一个改变另一个就会改变。3.torch.zeros():依size创建全0张量...原创 2020-01-13 21:22:01 · 1295 阅读 · 0 评论
分享