Struct Buffer 

pub struct Buffer {
    pub name: String,
    /* private fields */
}

A named non-learnable tensor buffer (e.g., running statistics).

Unlike Parameters, buffers are not tracked by optimizers but are persisted in checkpoints alongside parameters. Clone shares the underlying Rc, so the checkpoint system can write through to the same cell the owning module holds.

Fields

name: String

Implementations

impl Buffer

pub fn new(tensor: Tensor, name: &str) -> Self

Create a named buffer from a tensor.

pub fn get(&self) -> Tensor

Get a shallow clone of the underlying tensor.

pub fn set(&self, tensor: Tensor)

Replace the underlying tensor.

pub fn shape(&self) -> Vec<i64>

Shape of the underlying tensor.

pub fn device(&self) -> Device

Device of the underlying tensor.

pub fn to_device(&self, device: Device) -> Result<()>

Move buffer to a device (writes through the Rc).

Trait Implementations

impl Clone for Buffer

fn clone(&self) -> Buffer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Buffer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.