Pytorch张量通过索引获取指定数据

在这里插入图片描述

import torch
x = torch.tensor([1,2,3])

x
Out[3]: tensor([1, 2, 3])

x[0]  # 索引操作:取单个数字
Out[4]: tensor(1)

x[0:1]  # 切片操作:可以保持维度不变
Out[5]: tensor([1])

x[torch.tensor([True,False,True])]  # 布尔值索引,通过条件筛选,mask掩码
# 这个条件可以是tensor也可以是列表,只要长度一致
# x[[True,False,True]]
Out[6]: tensor([1, 3])

相关推荐

  1. pytorch中numel()函数用于获取中元素数目

    2024-01-13 16:02:01       25 阅读
  2. PyTorch

    2024-01-13 16:02:01       45 阅读
  3. Pytorch广播

    2024-01-13 16:02:01       36 阅读
  4. Pytorch笔记】

    2024-01-13 16:02:01       25 阅读
  5. PyTorch形状

    2024-01-13 16:02:01       20 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-13 16:02:01       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-13 16:02:01       101 阅读
  3. 在Django里面运行非项目文件

    2024-01-13 16:02:01       82 阅读
  4. Python语言-面向对象

    2024-01-13 16:02:01       91 阅读

热门阅读

  1. 如何控制docker服务的日志输出?

    2024-01-13 16:02:01       57 阅读
  2. 【WPF.NET开发】WPF中的XAML资源

    2024-01-13 16:02:01       45 阅读
  3. 如何正确了解应用高防IP

    2024-01-13 16:02:01       55 阅读
  4. C++的模板与泛型编程探秘

    2024-01-13 16:02:01       50 阅读
  5. Ubuntu 上升级nodejs 版本

    2024-01-13 16:02:01       64 阅读
  6. ubuntu篇---降低内核版本

    2024-01-13 16:02:01       60 阅读
  7. ubuntu重启后会修改挂载路径

    2024-01-13 16:02:01       65 阅读
  8. 删除文件夹及文件夹下的文件夹和文件

    2024-01-13 16:02:01       51 阅读
  9. LC 2085. 统计出现过一次的公共字符串

    2024-01-13 16:02:01       50 阅读
  10. 聊聊PowerJob的TimingStrategyHandler

    2024-01-13 16:02:01       48 阅读