✳flâneur — a map of the web's best reading

torch.ones_like — PyTorch 2.12 documentation

docs.pytorch.org · 236 words · saved by 1 readers

Returns a tensor filled with the scalar value 1, with the same size as input. torch.ones_like(input) is equivalent to torch.ones(input.size(), dtype=input.dtype, layout=input.layout, device=input.device). Warning As of 0.4, this function does not support an out keyword. As an alternative, the old torch.ones_like(input, out=output) is equivalent to torch.ones(input.size(), out=output). input (Tensor) – the size of input will determine size of the output tensor. dtype (torch.dtype, optional) – the desired data type of returned Tensor. Default: if None, defaults to the dtype of input. layout (torch.layout, optional) – the desired layout of returned tensor. Default: if None, defaults to the layout of input. device (torch.device, optional) – the desired device of returned tensor. Default: if None, defaults to the device of input. requires_grad (bool, optional) – If autograd should record operations on the returned tensor. Default: False. memory_format (torch.memory_format, optional) – the d

torch.ones_like # torch. ones_like ( input , * , dtype = None , layout = None , device = None , requires_grad = False , memory_format = torch.preserve_format ) → Tensor # Returns a tensor filled with the scalar value 1 , with the same size as input . torch.ones_like(input) is equivalent to torch.ones(input.size(), dtype=input.dtype, layout=input.layout, device=input.device) . Warning As of 0.4, this function does not support an out keyword. As an alternative, the old torch.ones_like(input, out=output) is equivalent to torch.ones(input.size(), out=output) . Parameters : input ( Tensor )

Explore this link on the map →

related reading