all and any do have function as well method variants which take dim

pytorch/aten/src/ATen/native/native_functions.yaml

Lines 471 to 498 in 4ed7f36

    
           - func: all.dim(Tensor self, int dim, bool keepdim=False) -> Tensor 
        
             use_c10_dispatcher: full 
        
             variants: function, method 
        
             dispatch: 
        
               CPU, CUDA: all 
        
           - func: all.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) 
        
             dispatch: 
        
               CPU, CUDA: all_out 
        
           - func: all.dimname(Tensor self, Dimname dim, bool keepdim=False) -> Tensor 
        
             variants: function, method 
        
           - func: all.dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) 
        
           - func: allclose(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_nan=False) -> bool 
        
             use_c10_dispatcher: full 
        
             variants: function, method 
        
           - func: any.dim(Tensor self, int dim, bool keepdim=False) -> Tensor 
        
             use_c10_dispatcher: full 
        
             variants: function, method 
        
             dispatch: 
        
               CPU, CUDA: any 
        
           - func: any.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!) 
        
             dispatch: 
        
               CPU, CUDA: any_out

They are not documented though.

	- func: all.dim(Tensor self, int dim, bool keepdim=False) -> Tensor
	use_c10_dispatcher: full
	variants: function, method
	dispatch:
	CPU, CUDA: all

	- func: all.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)
	dispatch:
	CPU, CUDA: all_out

	- func: all.dimname(Tensor self, Dimname dim, bool keepdim=False) -> Tensor
	variants: function, method

	- func: all.dimname_out(Tensor self, Dimname dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)

	- func: allclose(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_nan=False) -> bool
	use_c10_dispatcher: full
	variants: function, method

	- func: any.dim(Tensor self, int dim, bool keepdim=False) -> Tensor
	use_c10_dispatcher: full
	variants: function, method
	dispatch:
	CPU, CUDA: any

	- func: any.out(Tensor self, int dim, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)
	dispatch:
	CPU, CUDA: any_out

[FR] .all and .any should have dim argument #48352

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions