Apply suggestions from code review

This commit is contained in:
Patrick von Platen
2023-08-23 13:33:45 +02:00
committed by GitHub
parent c98d3cfd8b
commit 1f58bdbb1d
2 changed files with 3 additions and 3 deletions

View File

@ -31,7 +31,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
}
```
`cargo run` should display a tensor of shape `Tensor[[2, 4], f32]`
`cargo run` should display a tensor of shape `Tensor[[2, 4], f32]`.
Having installed `candle` with Cuda support, simply define the `device` to be on GPU: