n_data = torch.ones(5, 2) >>> print(n_data) tensor([[1., 1.] ... torch.max()返回两个结果,第一个是最大值,第二个是对应的索引值;第二个参数0 ...
確定! 回上一頁