fn new_layer_tensors(&self, batch_size: usize) -> LayerTensors<i32> {
        let layer_sizes = self.vectorizer.layer_sizes();

        let mut layers: EnumMap<Layer, TensorWrap<i32>> = EnumMap::new();
        for (layer, tensor) in &mut layers {
            *tensor = TensorWrap(Tensor::new(&[batch_size as u64, layer_sizes[layer] as u64]));
        }

        layers
    }
/// Construct net layer batch tensors.
///
/// Each tensor has shape `[batch_size, layer_size]`.
Construct net layer batch tensors.
Construct net layer batch tensors .
fn new_layer_tensors(&self, batch_size: usize) -> LayerTensors<i32> {
        let layer_sizes = self.vectorizer.layer_sizes();
        let mut layers: EnumMap<Layer, TensorWrap<i32>> = EnumMap::new();
        for (layer, tensor) in &mut layers {
            *tensor = TensorWrap(Tensor::new(&[batch_size as u64, layer_sizes[layer] as u64]));
        }
        layers
    }
fn new_layer_tensors ( & self , batch_size : usize ) -> LayerTensors < i32 > { let layer_sizes = self . vectorizer . layer_sizes ( ) ; let mut layers : EnumMap < Layer , TensorWrap < i32 > > = EnumMap :: new ( ) ; for ( layer , tensor ) in & mut layers { * tensor = TensorWrap ( Tensor :: new ( & [ batch_size as u64 , layer_sizes [ layer ] as u64 ] ) ) ; } layers }
Construct net layer batch tensors.
Construct net layer batch tensors .
&self
batch_size
usize
&self
batch_size
usize