Type Alias TensorGpu 

pub type TensorGpu<T, K> = Tensor<Gpu<K>, T>;

Aliased Type

pub struct TensorGpu<T, K> { /* private fields */ }

Implementations

impl<T: Scalar, K: Kind> TensorGpu<T, K>

pub fn from_data_u8( context: &Context, shape: impl Into<Shape>, contents: &[u8], ) -> Result<Self, TensorError>

pub fn back_in_place(&self) -> TensorCpu<T>

pub async fn back(&self) -> TensorCpu<T>

impl<T: Scalar, K: Kind> TensorGpu<T, K>

pub fn context(&self) -> &Context

pub fn load(&self, host: &TensorCpu<T>) -> Result<(), TensorError>

pub fn load_batch( &self, host: &TensorCpu<T>, batch: usize, ) -> Result<(), TensorError>

pub fn destroy(self)

impl<T: Scalar> TensorGpu<T, Uniform>

pub fn layout(&self, binding: u32) -> BindGroupLayoutEntry

impl<T: Scalar> TensorGpu<T, ReadWrite>

pub fn meta_layout(&self, binding: u32) -> BindGroupLayoutEntry

pub fn layout(&self, binding: u32, read_only: bool) -> BindGroupLayoutEntry

impl<T: Scalar> TensorGpu<T, ReadWrite>

pub fn view( &self, x: impl TensorAxis, y: impl TensorAxis, z: impl TensorAxis, w: impl TensorAxis, ) -> Result<TensorGpuView<'_, T>, TensorError>

Create a view for the tensor.

Trait Implementations

impl<T: Scalar> DeepClone for TensorGpu<T, ReadWrite>

fn deep_clone(&self) -> Self

impl<T: Scalar + Serialize, K: Kind> Serialize for TensorGpu<T, K>

Available on non-WebAssembly only.

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.

impl<T: Scalar, K: Kind> TensorInitContext<T> for TensorGpu<T, K>

fn from_data<'a, S, D>( context: &Context, shape: S, data: D, ) -> Result<Self, TensorError>

where S: Into<Shape>, D: Into<Cow<'a, [T]>>,

Init the tensor with given shape and contents.

fn init(context: &Context, shape: impl Into<Shape>) -> Self

Init the tensor with given shape.

impl<T: Scalar> TensorInto<Tensor<Gpu<ReadWrite>, T>> for TensorGpu<T, ReadWrite>

Available on non-WebAssembly only.

fn to(self, context: &Context) -> Self

impl<T: Scalar, K: Kind> TensorReshape for TensorGpu<T, K>

fn reshape( &self, x: TensorDimension, y: TensorDimension, z: TensorDimension, w: TensorDimension, ) -> Result<Self, TensorError>

impl<T: Scalar, K: Kind> TensorResource for TensorGpu<T, K>

fn resource_key(&self) -> ResourceKey

Retrieve the key identifying a resource.

fn meta_binding(&self) -> BindingResource<'_>

Binding for metadata of the tensor (shape, stride, etc.).

fn binding(&self) -> BindingResource<'_>

Binding for actual data of the tensor.

impl<T: Scalar, K: Kind> TensorShape for TensorGpu<T, K>

fn shape(&self) -> Shape

Get the shape of the tensor.

fn check_shape(&self, shape: impl Into<Shape>) -> Result<(), TensorError>

Check if the tensor’s shape is the same as what expected.

fn check_shape_any<S>(&self, shapes: &[S]) -> Result<(), TensorError>

where S: Into<Shape> + ToOwned<Owned = S>,

Check if the tensor’s shape matches any of the expected ones.