1. 首页
  2. 编程语言
  3. Python
  4. Pytorch学习之torch用法----比较操作(Comparison Ops)

Pytorch学习之torch用法----比较操作(Comparison Ops)

上传者: 2022-02-24 16:57:09上传 PDF文件 66.77 KB 热度 7次

说明: 比较元素是否相等,第二个参数可以是一个数,或者是第一个参数同类型形状的张量参数:input ---- 待比较张量other ---- 比较张量或者数out ---- 输出张量返回值: 一个torch.ByteTensor张量,包含了每个位置的比较结果说明: 如果两个张量有相同的形状和元素值,则返回true,否则False参数:tensor1 ---- 比较张量1tensor2 ---- 比较张量2out ---- 输出张量说明: 逐元素比较input和other,即是否input >= other。返回一个元组, 其中indices是原始输入张量中沿dim维的第k个最小值下标。如果指定参数descending为True,则按降序排序。如果不指定dim,则默认input的最后一维,如果largest为False,则返回最小的k个值。

用户评论