din

package

v0.4.0 Latest Latest Go to latest Published: Nov 1, 2022 License: AGPL-3.0 Imports: 7 Imported by: 0

Details

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

Repository

github.com/auxten/edgeRec

Links

Open Source Insights

Documentation ¶

Index ¶

func EucDistance(x, y *G.Node) (retVal *G.Node)
func PRelu(x, slop *G.Node) (retVal *G.Node)
type DinNet
- func NewDinNet(uProfileDim, uBehaviorSize, uBehaviorDim int, iFeatureDim int, cFeatureDim int) *DinNet
- func NewDinNetFromJson(data []byte) (din *DinNet, err error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func EucDistance ¶

func EucDistance(x, y *G.Node) (retVal *G.Node)

EucDistance is the Euclidean distance between two matrix, typically used for calculating the distance between two embedding. Case1: x, y shapes are same, no broadcast. output shape will be x.shape[:-1] Case2: x, y shapes are different, broadcast will be applied on the smaller dim. output shape will be something like x.shape[:-1] but with a broadcast dim

func PRelu ¶

func PRelu(x, slop *G.Node) (retVal *G.Node)

PRelu is the slop learnable LeakyRelu activation function slop should be a scalar

Types ¶

type DinNet ¶

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

func NewDinNet ¶

func NewDinNet(
	uProfileDim, uBehaviorSize, uBehaviorDim int,
	iFeatureDim int,
	cFeatureDim int,
) *DinNet

func NewDinNetFromJson ¶

func NewDinNetFromJson(data []byte) (din *DinNet, err error)

func (*DinNet) Fwd ¶

func (din *DinNet) Fwd(xUserProfile, xUbMatrix, xItemFeature, xCtxFeature *G.Node, batchSize, uBehaviorSize, uBehaviorDim int) (err error)

Fwd performs the forward pass xUserProfile: [batchSize, userProfileDim] xUbMatrix: [batchSize, uBehaviorSize* uBehaviorDim] xUserBehaviors: [batchSize, uBehaviorSize, uBehaviorDim] xItemFeature: [batchSize, iFeatureDim] xContextFeature: [batchSize, cFeatureDim]

func (*DinNet) Graph ¶

func (din *DinNet) Graph() *G.ExprGraph

func (*DinNet) In ¶

func (din *DinNet) In() G.Nodes

func (*DinNet) Learnable ¶ added in v0.4.0

func (din *DinNet) Learnable() G.Nodes

func (*DinNet) Marshal ¶

func (din *DinNet) Marshal() (data []byte, err error)

func (*DinNet) Out ¶

func (din *DinNet) Out() *G.Node

func (*DinNet) SetVM ¶

func (din *DinNet) SetVM(vm G.VM)

func (*DinNet) Vm ¶

func (din *DinNet) Vm() G.VM

Source Files ¶

View all Source files

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