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.