@@ -707,18 +707,18 @@ def take(x: array, indices: array, /, *, axis: Optional[int] = None, **kwargs) -
707
707
return torch .index_select (x , axis , indices , ** kwargs )
708
708
709
709
__all__ = ['result_type' , 'can_cast' , 'permute_dims' , 'bitwise_invert' ,
710
- 'newaxis' , 'add ' , 'atan2 ' , 'bitwise_and ' , 'bitwise_left_shift ' ,
711
- 'bitwise_or ' , 'bitwise_right_shift ' , 'bitwise_xor' , 'copysign ' ,
712
- 'divide ' , 'equal ' , 'floor_divide ' , 'greater ' , 'greater_equal ' ,
713
- 'less ' , 'less_equal ' , 'logaddexp ' , 'multiply ' , 'not_equal' , 'pow ' ,
714
- 'remainder ' , 'subtract ' , 'max ' , 'min ' , 'clip ' , 'sort' , 'prod ' ,
715
- 'sum ' , 'any ' , 'all ' , 'mean ' , 'std ' , 'var ' , 'concat ' , 'squeeze ' ,
716
- 'broadcast_to ' , 'flip ' , 'roll ' , 'nonzero ' , 'where ' , 'reshape ' ,
717
- 'arange ' , 'eye ' , 'linspace ' , 'full ' , 'ones ' , 'zeros ' , 'empty ' ,
718
- 'tril ' , 'triu ' , 'expand_dims ' , 'astype ' , 'broadcast_arrays ' ,
719
- 'UniqueAllResult ' , 'UniqueCountsResult ' , 'UniqueInverseResult ' ,
720
- 'unique_all ' , 'unique_counts ' , 'unique_inverse' , 'unique_values ' ,
721
- 'matmul ' , 'matrix_transpose ' , 'vecdot ' , 'tensordot' , 'isdtype ' ,
722
- 'take' ]
710
+ 'newaxis' , 'conj ' , 'add ' , 'atan2 ' , 'bitwise_and ' ,
711
+ 'bitwise_left_shift ' , 'bitwise_or ' , 'bitwise_right_shift ' ,
712
+ 'bitwise_xor ' , 'copysign ' , 'divide ' , 'equal ' , 'floor_divide ' ,
713
+ 'greater ' , 'greater_equal ' , 'less ' , 'less_equal ' , 'logaddexp ' ,
714
+ 'multiply ' , 'not_equal ' , 'pow ' , 'remainder ' , 'subtract ' , 'max ' ,
715
+ 'min ' , 'clip ' , 'sort ' , 'prod ' , 'sum ' , 'any ' , 'all ' , 'mean' , 'std ' ,
716
+ 'var ' , 'concat ' , 'squeeze ' , 'broadcast_to ' , 'flip ' , 'roll ' ,
717
+ 'nonzero ' , 'where ' , 'reshape ' , 'arange ' , 'eye ' , 'linspace ' , 'full ' ,
718
+ 'ones ' , 'zeros ' , 'empty ' , 'tril ' , 'triu' , 'expand_dims' , 'astype ' ,
719
+ 'broadcast_arrays ' , 'UniqueAllResult ' , 'UniqueCountsResult ' ,
720
+ 'UniqueInverseResult ' , 'unique_all ' , 'unique_counts ' ,
721
+ 'unique_inverse ' , 'unique_values ' , 'matmul ' , 'matrix_transpose ' ,
722
+ 'vecdot' , 'tensordot' , 'isdtype' , ' take' ]
723
723
724
724
_all_ignore = ['torch' , 'get_xp' ]
0 commit comments