Convert DType.uint256 to UInt256?

Mojo
1

Can anyone guide me convert a DType.uint256 number to UInt256?
I failed to convert DType.uint256 to UInt256 with this code:

comptime layout = Layout.row_major(5)
tensor = LayoutTensor[DType.uint256, layout, MutAnyOrigin].stack_allocation()
tensor[0] = Scalar[DType.uint256](10)
var host_scalar: Scalar[DType.uint256] = tensor[0]
print("host_scalar =", host_scalar)

Error:

error: cannot implicitly convert 'LayoutTensor[DType.uint256, layout, MutAnyOrigin].element_type' value to 'UInt256'
    var host_scalar: Scalar[DType.uint256] = tensor[0]
                                             ~~~~~~^~~
2

The issue is that LayoutTensor.__getitem__ returns SIMD[Self.dtype, Self.element_size] (in order to handle vectorization). You can get tensor[0][0] (the first element of the parametric SIMD type). Due to the way that element_size is defined, the compiler will not be able to deduce that element_size=1 in the near future. There is also LayoutTensor.load_scalar() as another option.