fcCovdwd: Covariance between derivatives and weights*derivatives

In mgoulet847/tagi: Tractable Approximate Gaussian Inference in Neural Networks

Description

This function calculates covariance between derivatives and weights and covariance between derivatives from consecutive layers.

Usage

fcCovdwd(md, mw, Cdow, Cdodi, ni, no, B)

Arguments

md	Mean vector of derivatives
mw	Mean vector of weights for the current layer
Cdow	Covariance between derivatives and weights
Cdodi	Covariance between derivatives from current and next layers
ni	Number of units in current layer
no	Number of units in next layer
B	Batch size

Value

Covariance between derivatives and weights times derivatives

mgoulet847/tagi documentation built on Dec. 21, 2021, 5:10 p.m.