- Tensor.type(dtype=None, non_blocking=False, **kwargs) str or Tensor ¶
Returns the type if dtype is not provided, else casts this object to the specified type.
If this is already of the correct type, no copy is performed and the original object is returned.
dtype (dtype or string) – The desired type
non_blocking (bool) – If
True, and the source is in pinned memory and destination is on the GPU or vice versa, the copy is performed asynchronously with respect to the host. Otherwise, the argument has no effect.
**kwargs – For compatibility, may contain the key
asyncin place of the
asyncarg is deprecated.