(self, shape, fill_value, type_as=None)
| 2005 | return torch.arange(start, stop, step, device=type_as.device) |
| 2006 | |
| 2007 | def full(self, shape, fill_value, type_as=None): |
| 2008 | if isinstance(shape, int): |
| 2009 | shape = (shape,) |
| 2010 | if type_as is None: |
| 2011 | return torch.full(shape, fill_value) |
| 2012 | else: |
| 2013 | return torch.full( |
| 2014 | shape, fill_value, dtype=type_as.dtype, device=type_as.device |
| 2015 | ) |
| 2016 | |
| 2017 | def eye(self, N, M=None, type_as=None): |
| 2018 | if M is None: |