@@ -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
@@ -694,7 +695,7 @@ def take(x: array, indices: array, /, *, axis: Optional[int] = None, **kwargs) -
694
695
'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
- 'logaddexp' , 'multiply' , 'not_equal' , 'pow' , 'remainder' ,
698
+ 'logaddexp' , 'multiply' , 'newaxis' , ' not_equal' , 'pow' , 'remainder' ,
698
699
'subtract' , 'max' , 'min' , 'sort' , 'prod' , 'sum' , 'any' , 'all' ,
699
700
'mean' , 'std' , 'var' , 'concat' , 'squeeze' , 'broadcast_to' , 'flip' , 'roll' ,
700
701
'nonzero' , 'where' , 'reshape' , 'arange' , 'eye' , 'linspace' , 'full' ,
0 commit comments