diff --git a/source/machine-learning/pytorch.md b/source/machine-learning/pytorch.md index c68c127..df4333a 100644 --- a/source/machine-learning/pytorch.md +++ b/source/machine-learning/pytorch.md @@ -2,4 +2,7 @@ ## Commands -- [torch.Tensor.item()](https://pytorch.org/docs/stable/generated/torch.Tensor.item.html): convert tensor to standard Python number - only works for tensors with one element +- [torch.Tensor.item()](https://pytorch.org/docs/stable/generated/torch.Tensor.item.html) + - convert tensor to standard Python number + - only works for tensors with one element + - implicitly moves the value to the CPU