huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Switch Tensor::full to return a contiguous tensor. (#2929)

Browse Source
This commit is contained in:
Laurent Mazare
2025-04-28 21:36:39 +02:00
committed by GitHub
parent d4bac37a61
commit de23d34a28
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -382,8 +382,7 @@ impl Tensor {
Self::new_impl(array, shape, device, false) Self::new_impl(array, shape, device, false)
} }
/// Returns a new tensor with all the elements having the same specified value. Note that /// Returns a new tensor with all the elements having the same specified value.
/// the tensor is not contiguous so you would have to call `.contiguous()` on it if needed.
///```rust ///```rust
/// use candle_core::{Tensor, Device}; /// use candle_core::{Tensor, Device};
/// let a = Tensor::full(3.5, (2, 4), &Device::Cpu)?; /// let a = Tensor::full(3.5, (2, 4), &Device::Cpu)?;
@ -398,7 +397,12 @@ impl Tensor {
shape: S, shape: S,
device: &Device, device: &Device,
) -> Result<Self> { ) -> Result<Self> {
Self::from_vec_impl(vec![value], (), device, false)?.broadcast_as(shape) let none = BackpropOp::none();
let shape = shape.into();
let mut storage = unsafe { device.alloc_uninit(&shape, D::DTYPE)? };
let layout = Layout::contiguous(shape.clone());
storage.const_set(value.to_scalar(), &layout)?;
Ok(from_storage(storage, shape, none, false))
} }
/// Creates a new 1D tensor from an iterator. /// Creates a new 1D tensor from an iterator.