torch.register_dtype
function register_dtype<const Name extends string, const Components extends number, ComponentType extends DType>(config: DTypeConfig<Name, Components, ComponentType>): DTypeHandleImpl<Name>function register_dtype<const Name extends string, const Components extends number, ComponentType extends DType>(name: Name, components: Components, componentType: ComponentType, zero: TupleOfLength<number, Components>, options?: RegisterDTypeOptions): DTypeHandleImpl<Name>Register a custom dtype with full configuration object.
Parameters
configDTypeConfig<Name, Components, ComponentType>- Full dtype configuration
Returns
DTypeHandleImpl<Name>– Handle to the registered dtypeExamples
torch.library.register_dtype({
name: 'quaternion_f32',
components: 4,
componentType: 'float32',
zero: [1, 0, 0, 0],
promotes_from: ['float32'],
});