T in (tensor(bfloat16),Β tensor(double),Β tensor(float),Β tensor(float16),Β tensor(int32),Β tensor(int64),Β tensor(uint32),
tensor(uint64)) : Constrain input and output types to high-precision numeric tensors.

 

T2 in (tensor(int32),Β tensor(int64)) : axis tensor can be int32 or int64 only.