pymoose.cast#

pymoose.cast(x, dtype, placement=None)[source]#

Cast a tensor to a new dtype.

Parameters
  • x – A tensor.

  • dtype – A DType.

  • placement – An optional Placement to pin this operation to.

Returns

The tensor x converted to dtype.