torch

package module

v0.0.0-...-89e5866 Latest Latest Go to latest Published: Feb 5, 2026 License: AGPL-3.0 Imports: 4 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/Jimmy2099/torch

Links

Open Source Insights

README ¶

BioTorch - Neuromorphic Compute(Bio-Neuron Inspired) DeepLearning Engine (WIP)

Inference Compute Runtime Support (WIP)

ONNX Runtime (Version 7 Export Only)
WGPU (WIP)

llama3 and llama3 vision example built on top of this framework

CNN From Scratch example AND re-design build-in Graphic Renderer that compatible with Tensor data struct:

Clear Structure For Highly Customizable Research AND Compute Graph and Visualization:

AutoEncoder Compute Graph and Visualization

More Examples: README.md

Notice

Some parts of this project’s code may generate, optimize, or annotate by AI.

Documentation ¶

Index ¶

type Embedding
- func NewEmbedding(vocabSize, embDim int) *Embedding
type Layer
type LayerForTesting
type LayerLoader

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type Embedding ¶

type Embedding struct {
	Weights     *tensor.Tensor
	GradWeights *tensor.Tensor
	VocabSize   int
	EmbDim      int
	LastIndices []int
}

func NewEmbedding ¶

func NewEmbedding(vocabSize, embDim int) *Embedding

func (*Embedding) Backward ¶

func (e *Embedding) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor

func (*Embedding) Forward ¶

func (e *Embedding) Forward(indices *tensor.Tensor) *tensor.Tensor

func (*Embedding) GetBias ¶

func (e *Embedding) GetBias() *tensor.Tensor

func (*Embedding) GetWeights ¶

func (e *Embedding) GetWeights() *tensor.Tensor

func (*Embedding) Parameters ¶

func (e *Embedding) Parameters() []*tensor.Tensor

func (*Embedding) SetBiasAndShape ¶

func (e *Embedding) SetBiasAndShape(data []float32, shape []int)

func (*Embedding) SetWeightsAndShape ¶

func (e *Embedding) SetWeightsAndShape(data []float32, shape []int)

func (*Embedding) ZeroGrad ¶

func (e *Embedding) ZeroGrad()

type Layer ¶

type Layer interface {
	Forward(input *tensor.Tensor) *tensor.Tensor

	//DEL
	Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor
	ZeroGrad()
	Parameters() []*tensor.Tensor
}

type LayerForTesting ¶

type LayerForTesting interface {
	GetWeights() *tensor.Tensor
	GetBias() *tensor.Tensor
}

type LayerLoader ¶

type LayerLoader interface {
	SetWeightsAndShape(data []float32, shape []int)
	SetBiasAndShape(data []float32, shape []int)
}

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
cmd
07_autoencoder_compute_graph command
08_complier_testing command
data_store
compute_dependency_graph
compute_graph
data_type
memory
network
neuromorphic_compute
node
tensor
datasets_loader
mnist
zinc
layer
layer_cg
legacy
00_standalone_example command
01_polynomial_regression command
02_cnn_mnist_handwrite_recognize command
03_autoencoder command
04_variational_autoencoder command
04_variational_autoencoder/py command
06_layer_testing command
pkg
algorithm
common
fmt
graph_visualize
log
risc-v
rv64-core/cmd/vm command
rv64-core/pkg/rv64
rv64-core/pkg/rv64/cmd/make command
rv64-core/pkg/rv64/cmd/rv64 command
testing
thirdparty
float16
fp16_convert Package float16 defines support for half-precision floating-point numbers.	Package float16 defines support for half-precision floating-point numbers.
onnx-go
onnx-go/backend
onnx-go/backend/simple Package simple holds a very simple graph structure suitable to receive an onnx model	Package simple holds a very simple graph structure suitable to receive an onnx model
onnx-go/backend/testbackend Package testbackend provides a set of testing helper functions that test backend interface implementations.	Package testbackend provides a set of testing helper functions that test backend interface implementations.
onnx-go/backend/testbackend/onnx Package onnxtest contains an export of the onnx test files	Package onnxtest contains an export of the onnx test files
onnx-go/backend/testbackend/onnx/gen_cmd command
onnx-go/backend/testbackend/testreport
onnx-go/backend/x/gorgonnx Package gorgonnx creates a temporary graph that is compatible with backend.ComputationBackend.	Package gorgonnx creates a temporary graph that is compatible with backend.ComputationBackend.
onnx-go/backend/x/gorgonnx/genapi command
onnx-go/doc/introduction command Present displays slide presentations and articles.	Present displays slide presentations and articles.
onnx-go/doc/introduction/samples command
onnx-go/doc/introduction/utils command
onnx-go/examples/emotions command
onnx-go/examples/model_zoo_executor command
onnx-go/examples/tiny_yolov2 command
onnx-go/internal/examples/mnist
onnx-go/internal/examples/mnist/gen command
onnx-go/internal/onnx/ir
onnx-go/internal/readme
onnx-go/internal/tools/dump command
onnx-go/internal/x/images
onnx-go/ir
onnx-go/misc/viewer command
tiktoken
tiktoken/cmd/tokenizer command
tiktoken/codec
tiktoken/internal/cmd command
vision
image/png
image/raw3d
screen

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL