vector

package

v0.0.2 Latest Latest Go to latest Published: May 11, 2023 License: MIT Imports: 4 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Vector3D struct {
	// contains filtered or unexported fields
}

func NewVector3D(batch, rows, cols int) *Vector3D

func ReshapeMatrix(d mat.Matrix, rows, cols int) *Vector3D

func ReshapeVector(vec mat.Vector, rows, cols int) *Vector3D

func (v *Vector3D) BatchSize() int

func (v *Vector3D) ConvAdd(a *Vector3D, strideM, strideN int)

ConvAdd add gradient to each channel

func (v *Vector3D) Cut(rows, cols int) *Vector3D

Cut cut matrix to (rows, cols)

func (v *Vector3D) Dims() (int, int)

func (v *Vector3D) Get(batch int) mat.Matrix

func (v *Vector3D) Im2Col(kernelM, kernelN, strideM, strideN, channelSize int) *mat.Dense

Im2Col get convolution matrix, output shape is (batch * conved.rows * conved.cols, kernelM * kernelN * channelSize)

func (v *Vector3D) Pad(m, n int)

Pad padding matrix on right and bottom

func (v *Vector3D) ToMatrix() mat.Matrix