HowToSD-ComfyUI-Pt-Wrapper/docs/reference/pt_argmax.md

Pt Argmax

Computes the indices of the maximum values of a PyTorch tensor along the specified dimension(s).

Specify the dimension in the dim field using an integer as shown below:

0

Input

Name	Data type
tens	Tensor
dim	String
keepdim	Boolean

Output

Data type
Tensor

---

Category: PyTorch wrapper - Reduction operation & Summary statistics

ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.