Documentation ¶
Index ¶
- func ArgMaxCol(x Tensor) [][]int
- func Pad4D(input [][][][]float64, pad [][]int) [][][][]float64
- func Reshape2D1D(x Tensor) []float64
- func Reshape2D6D(input [][]float64, reN int, reC int, reH int, reW int, reX int, reY int) [][][][][][]float64
- func Reshape4D6D(input [][][][]float64, reN int, reC int, reH int, reW int, reX int, reY int) [][][][][][]float64
- func Shape2D(x Tensor) (n int, c int)
- func Shape4D(input Tensor) (n int, c int, h int, w int)
- func Shape6D(input [][][][][][]float64) (n int, c int, h int, w int, x int, y int)
- type Tensor
- func Add(x, y Tensor) Tensor
- func AddE(x Tensor, y float64) Tensor
- func Apply(x Tensor, fn func(float64) float64) Tensor
- func Cast(x Tensor, castSize int) Tensor
- func Conv1D(x, filter Tensor, stride int) Tensor
- func Div(x, y Tensor) Tensor
- func Dot(x, y Tensor) Tensor
- func HeNorm2D(r int, c int) Tensor
- func Make(shape []int) Tensor
- func Make2DInitArray(x [][]float64) Tensor
- func MakeInit(n int, m int, value float64) Tensor
- func MaxCol(x Tensor) Tensor
- func Mul(x, y Tensor) Tensor
- func MulE(x Tensor, y float64) Tensor
- func RandomNorm2D(r int, c int, init float64) Tensor
- func Reshape1D2D(x []float64, n, c int) Tensor
- func Reshape4D(input Tensor, reX int, reY int) Tensor
- func Sub(x, y Tensor) Tensor
- func SubE(x Tensor, y float64) Tensor
- func SumCol(x Tensor) Tensor
- func SumRow(x Tensor) Tensor
- func T(x Tensor) Tensor
- func Trans2D(input Tensor, n int, c int) Tensor
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func Reshape2D1D ¶
func Reshape2D6D ¶
func Reshape4D6D ¶
Types ¶
Click to show internal directories.
Click to hide internal directories.