singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "wangwei (JIRA)" <j...@apache.org>
Subject [jira] [Resolved] (SINGA-198) Change Layer::Setup API to include input Tensor shapes
Date Tue, 28 Jun 2016 03:52:57 GMT

     [ https://issues.apache.org/jira/browse/SINGA-198?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]

wangwei resolved SINGA-198.
---------------------------
    Resolution: Fixed
      Assignee: wangwei

> Change Layer::Setup API to include input Tensor shapes
> ------------------------------------------------------
>
>                 Key: SINGA-198
>                 URL: https://issues.apache.org/jira/browse/SINGA-198
>             Project: Singa
>          Issue Type: Improvement
>            Reporter: wangwei
>            Assignee: wangwei
>
> For some layers, they require input Tensor shapes to setup their own data structures,
e.g., parameters.
> Previously, we get those shape data from layer configuration (i.e., the protobuf object).
This ticket is going to pass the input Tensor shapes as an argument to the Layer::Setup function.
There are two cases:
> 1. there is a single input Tensor, then shape is a vector<size_t>
> 2. there are multiple input Tensor, then shape is vector<vector<size_t>>
> During setup, we may not know the batchsize, hence we use the shapes  for one sample/example/instance.
In other words, batchsize is not included in the shapes. E.g., the shape for convoltuion::Setup
has three values for channel, height and width, while batchsize is not included.
> Corresponding, we need a get_output_shapes function, which returns the shape of the features
generated by this layer. It should be called after calling Layer::Setup.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

Mime
View raw message