huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Put them back.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-10 15:11:48 +02:00
parent 38ac50eeda
commit 49f4a77ffd
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
candle-core/src/tensor.rs
View File

@ -156,11 +156,11 @@ impl Tensor {
// Hiding it from now, having this functions forces us to have *every* function that creates
// a new tensor potentially `_var` Maybe having something more like `Tensor::ones(..).var()`
// might be easier to check.
// pub fn ones_var<S: Into<Shape>>(shape: S, dtype: DType, device: &Device) -> Result<Self> {
// // Maybe we should allocate some actual storage for vars rather than just using a
// // broadcasted scalar?
// Self::ones_impl(shape, dtype, device, true)
// }
pub fn ones_var<S: Into<Shape>>(shape: S, dtype: DType, device: &Device) -> Result<Self> {
// Maybe we should allocate some actual storage for vars rather than just using a
// broadcasted scalar?
Self::ones_impl(shape, dtype, device, true)
}
/// Create a new tensors filled with ones with same shape, dtype, and device
/// as the other tensors
@ -208,9 +208,9 @@ impl Tensor {
Self::zeros_impl(shape, dtype, device, false)
}
// pub fn zeros_var<S: Into<Shape>>(shape: S, dtype: DType, device: &Device) -> Result<Self> {
// Self::zeros_impl(shape, dtype, device, true)
// }
pub fn zeros_var<S: Into<Shape>>(shape: S, dtype: DType, device: &Device) -> Result<Self> {
Self::zeros_impl(shape, dtype, device, true)
}
/// Create a new tensors filled with ones with same shape, dtype, and device
/// as the other tensors