Trait border_tch_agent::ZeroTensor

pub trait ZeroTensor {
    // Required method
    fn zeros(shape: &[i64]) -> Tensor;
}

Adds capability of constructing Tensor with a static method.

Required Methods

fn zeros(shape: &[i64]) -> Tensor

Constructs zero tensor.

Implementations on Foreign Types

impl ZeroTensor for f32

fn zeros(shape: &[i64]) -> Tensor

impl ZeroTensor for i32

fn zeros(shape: &[i64]) -> Tensor

impl ZeroTensor for u8

fn zeros(shape: &[i64]) -> Tensor

impl ZeroTensor for i64

fn zeros(shape: &[i64]) -> Tensor

Implementors