首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

tensors used as indices must be long or byte tensors

这样做会将索引张量的数据类型转换为要索引的张量相匹配的类型。...1, 0, 1])# 选择标签为1的数据进行处理selected_data = dataset[labels == 1] # 使用张量作为索引# 打印选中的数据print(selected_data...具体而言,参数说明如下:​​dim​​:一个整数,表示要在哪个维度上进行索引选择。该值必须在输入张量的有效范围内。​​index_tensor​​:一个包含索引值的一维整型张量。...该张量的形状必须输入张量中​​dim​​维度的大小相同,或者可以广播到与其大小相同。...对于按行选择,我们传递参数​​dim=0​​表示按行进行索引选择,​​index=torch.tensor([0, 2])​​是一个包含索引值的一维张量,它表示我们要选择输入张量中的第0行和第2行。

31730
领券