Struct ArcTensor 

pub struct ArcTensor(pub Arc<ReentrantMutex<RefCell<Tensor>>>);

原子引用计数（线程安全）张量句柄。

Tuple Fields

0: Arc<ReentrantMutex<RefCell<Tensor>>>

Implementations

impl ArcTensor

pub fn lock(&self) -> ReentrantMutexGuard<'_, RefCell<Tensor>>

获取内部数据的互斥锁守卫

pub fn from_tensor(t: Tensor) -> Self

从 Tensor 创建新的 ArcTensor

pub fn into_inner(self) -> Arc<ReentrantMutex<RefCell<Tensor>>>

获取内部句柄的克隆

impl ArcTensor

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

pub fn dtype(&self) -> DType

pub fn device(&self) -> Device

impl ArcTensor

pub fn to_vec<T: Pod + DTypeMapping>(&self) -> Result<Vec<T>>

pub fn from_vec<T: Pod + DTypeMapping>( data: Vec<T>, shape: Vec<usize>, device: Device, ) -> Result<Self>

pub fn from_scalar<T: Pod + DTypeMapping>( value: T, device: Device, ) -> Result<Self>

impl ArcTensor

pub fn zeros(shape: Vec<usize>, dtype: DType, device: Device) -> Result<Self>

创建指定形状和数据类型的零张量

pub fn ones(shape: Vec<usize>, dtype: DType, device: Device) -> Result<Self>

创建指定形状和数据类型的全一张量

pub fn empty(shape: Vec<usize>, dtype: DType, device: Device) -> Result<Self>

创建指定形状和数据类型的空张量（未初始化，当前为零初始化）

impl ArcTensor

pub fn as_view(&self) -> ArcTensorView

获取内部句柄的视图

pub fn broadcast_to(&self, target_shape: &[usize]) -> Result<ArcTensorView>

pub fn transpose(&self, axes: &[usize]) -> Result<ArcTensorView>

pub fn T(&self) -> Result<ArcTensorView>

impl ArcTensor

pub fn fill<T: Pod + DTypeMapping>(&mut self, value: T) -> Result<()>

Trait Implementations

impl Add for ArcTensor

type Output = ArcTensor

The resulting type after applying the + operator.

fn add(self, other: Self) -> Self::Output

Performs the + operation.

impl AddAssign for ArcTensor

fn add_assign(&mut self, other: Self)

Performs the += operation.

impl AsView for ArcTensor

type View = ArcTensorView

fn as_view(&self) -> Self::View

impl Clone for ArcTensor

fn clone(&self) -> ArcTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ArcTensor

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

Formats the value using the given formatter.

impl From<Tensor> for ArcTensor

fn from(t: Tensor) -> Self

Converts to this type from the input type.