import torch
ls = torch.Tensor([[0.2,0.4],
[0.3,0.2]])
print(ls.max(1))
代码理解 pseudo_labeled = outputs.max(1)[1]
PyTorch中的张量操作：max(1)函数实例
本文展示了如何在PyTorch中使用`torch.Tensor`类的`max(1)`方法来获取一维张量中每个元素的最大值，突出了深度学习库中的基础张量操作。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
