Struct TensorPatterns 

pub struct TensorPatterns;

Helper for creating common tensor patterns

Implementations

impl TensorPatterns

pub fn weight_matrix( name: String, input_dim: u64, output_dim: u64, tensor_type: TensorType, data: Vec<u8>, ) -> Result<(TensorInfo, TensorData)>

Create a weight matrix tensor

pub fn bias_vector( name: String, dim: u64, tensor_type: TensorType, data: Vec<u8>, ) -> Result<(TensorInfo, TensorData)>

Create a bias vector tensor

pub fn embedding_matrix( name: String, vocab_size: u64, embedding_dim: u64, tensor_type: TensorType, data: Vec<u8>, ) -> Result<(TensorInfo, TensorData)>

Create an embedding matrix