torch.max: 参数详解

torch.max(input) → Tensor

Parameters

  • input (Tensor) – the input tensor.

  • dim (int) – the dimension to reduce.

  • keepdim (bool) – whether the output tensor has dim retained or not. Default: False.

Keyword Arguments

        out (tupleoptional) – the result tuple of two output tensors (max, max_indices)

你可能感兴趣的:(Python,torch)