Currently only F32 is supported. Would like to support all datatypes especially uint16, f16, bf16, complex64 and complex128. uint16 isn't currently supported by pytorch. I've added a note in support of it's inclusion in https://github.com/pytorch/pytorch/issues/58734
Currently only F32 is supported.
Would like to support all datatypes especially uint16, f16, bf16, complex64 and complex128.
uint16 isn't currently supported by pytorch. I've added a note in support of it's inclusion in pytorch/pytorch#58734