pred = y_.max(-1, keepdim=True)[1]
时间: 2023-05-15 07:07:23 浏览: 69
Code.rar_PRED-163_matlab pred_社交网络_社交网络分析 链路预测_链路预测
5星 · 资源好评率100%
这是一个 PyTorch 中的代码,用于获取张量 y_ 中每行最大值的索引。具体来说,它返回一个张量,其中每个元素都是 y_ 中对应行的最大值的索引。keepdim=True 表示保持输出张量的维度与输入张量相同。
阅读全文
Code.rar_PRED-163_matlab pred_社交网络_社交网络分析 链路预测_链路预测