@@ -144,6 +144,7 @@ def can_cast(from_: Union[Dtype, array], to: Dtype, /) -> bool:
144
144
145
145
# Basic renames
146
146
bitwise_invert = torch .bitwise_not
147
+ newaxis = None
147
148
148
149
# Two-arg elementwise functions
149
150
# These require a wrapper to do the correct type promotion on 0-D tensors
@@ -690,8 +691,8 @@ def take(x: array, indices: array, /, *, axis: Optional[int] = None, **kwargs) -
690
691
axis = 0
691
692
return torch .index_select (x , axis , indices , ** kwargs )
692
693
693
- __all__ = ['result_type' , 'can_cast' , 'permute_dims' , 'bitwise_invert' , 'add ' ,
694
- 'atan2' , 'bitwise_and' , 'bitwise_left_shift' , 'bitwise_or' ,
694
+ __all__ = ['result_type' , 'can_cast' , 'permute_dims' , 'bitwise_invert' , 'newaxis ' ,
695
+ 'add' , ' atan2' , 'bitwise_and' , 'bitwise_left_shift' , 'bitwise_or' ,
695
696
'bitwise_right_shift' , 'bitwise_xor' , 'divide' , 'equal' ,
696
697
'floor_divide' , 'greater' , 'greater_equal' , 'less' , 'less_equal' ,
697
698
'logaddexp' , 'multiply' , 'not_equal' , 'pow' , 'remainder' ,
0 commit comments