torch多维取数据

最新推荐文章于 2024-03-23 23:21:08 发布

AI算法网奇

最新推荐文章于 2024-03-23 23:21:08 发布

阅读量3.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/82703153

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

介绍PyTorch中如何对多维数据进行索引操作，强调索引类型限制为uint8或int64，同时展示了使用该类型索引后的结果示例：12, 3。" 102681307,7574987,QML 实战：获取TableView选中行数据,"['QML开发', '用户界面', '数据展示', '事件处理', '表格操作']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

索引类型只支持uint8或者int64

bytes 或者long类型

import torch
import numpy as np
a=torch.zeros((4,3,3))

b=torch.ones((4,3)).type(torch.uint8)

print(a[b])

结果是12,3

# import numpy as np # # a = np.zeros((80,10, 3)) # # a=np.array([1,2,3]) # b = np.zeros((80,10),dtype=np.int32) # # print(a[b]) import torch import numpy as np a=torch.zeros((4,3)) b=np.asarray([[1,2],[1,2],[0,1]]) b=torch.ones((4,3)).type(torch.uint8) # b+=1 c=[1,2] print(a[b]) a[a<=0]=1 print(a)