How to create Multiplication Layer in deep learning toolbox

조회 수: 3 (최근 30일)
Jeong-Won Jeong
Jeong-Won Jeong 2018년 12월 19일
댓글: Abdelwahab Afifi 2020년 7월 22일
Dear community,
I am trying to create multiplication layer which multiplies inputs from multiple neural network layers element-wise, something like "additionLayer to add inputs from multiple layers element-wise". Is there any way I can create this layer in Matlab 2018a?
Best Regards,

답변 (1개)

CARLOS VIDAL
CARLOS VIDAL 2019년 10월 3일
편집: CARLOS VIDAL 2019년 10월 4일
Hi,
I don't think matlab already has available the layer you are looking for, but you can create one.
Follow the link below.
I did it quickly, it seems to work, but better if you double check and follow the instruction in the link above to know how to integrate it with your NN structure.
Hope this will help you.
classdef ElementWiseMultiplication < nnet.layer.Layer
% Example custom ElementWiseMultiplication layer.
properties (Learnable)
% Layer learnable parameters
% Scaling coefficients
end
methods
function layer = ElementWiseMultiplication(numInputs,name)
% layer = ElementWiseMultiplication(numInputs,name) creates a
% element wise multiplication and specifies the number of inputs
% and the layer name.
% Set number of inputs.
layer.NumInputs = numInputs;
% Set layer name.
layer.Name = name;
% Set layer description.
layer.Description = "Element Wise Multiplication of " + numInputs + ...
" inputs";
end
function Z = predict(~, X1,X2)
% Z = predict(layer, X1, ..., Xn) forwards the input data X1,
% ..., Xn through the layer and outputs the result Z.
% Element Wise Multiplication
Z = X1 .*X2;
end
end
end
  댓글 수: 2
ruo li
ruo li 2020년 4월 28일
Do I need to define a backward function?Thanks.
Abdelwahab Afifi
Abdelwahab Afifi 2020년 7월 22일
Have you get such layer? If yes, please share the code.

댓글을 달려면 로그인하십시오.

카테고리

Help CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by