logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

关于torch.nonzero() 详解

官方介绍: “返回一个包含输入 input 中非零元素索引的张量.输出张量中的每行包含 input 中非零元素的索引. 如果输入 input 有 n 维,则输出的索引张量 out 的 size 为 z x n , 这里 z 是输入张量 input 中所有非零元素的个数.”其实连着例子看的话,第一句话是很好理解的,就是第二句话有点搞不清楚究竟想说什么。例子>>> torch.non

#pytorch#python#深度学习
暂无文章信息