tensor.max() 在不同维度比较大小,并取最大值. # 给定一个三维矩阵--[batch_size, n_tags, n_tags] import torch score = torch.tensor([[ [1,2,7], ...
確定! 回上一頁