diff --git a/source/machine-learning/pytorch.md b/source/machine-learning/pytorch.md index df4333a..2c65971 100644 --- a/source/machine-learning/pytorch.md +++ b/source/machine-learning/pytorch.md @@ -4,5 +4,5 @@ - [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 + - only works for tensors with one element - otherwise an exception is raised - implicitly moves the value to the CPU