Pytorch

Pytorch

torch

• judgement

  torch.is_tensor(obj)
  torch.is_storage(obj)
  torch.is_complex(obj)
  torch.is_floating_point(obj)
  torch.is_nonzero(obj)

• dtype

  torch.set_default_dtype(obj)
  torch.get_default_dtype(obj)

• data from numpy

  torch.from_numpy(ndarray)