Developer Guide for Intel® Data Analytics Acceleration Library 2019 Update 2
The forward fully-connected layer computes values
for n input arguments x k , weights w ki , weights mask s ki , and biases b i , where k ∈ {1, ..., n}, i ∈ {1, ..., m}, and m is the number of layer outputs.
Given:
Dimension k
p-dimensional tensor X of size n 1 x ... x n k ... x n p
p-dimensional tensor W of size n 1 x ... x n k-1 x m x n k+1 ... x n p
p-dimensional tensor S of size n 1 x ... x n k-1 x m x n k+1 ... x n p
1-dimensional tensor B of size m