src/test/scripts/functions/tensor/Conv1DBackwardDataTest.dml

#-------------------------------------------------------------
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
#
#-------------------------------------------------------------

source("scripts/nn/layers/conv1d.dml") as conv1d

# Build input
rowBuild = seq(1,$numChannels*$imgSize,$numChannels)
if ( $numChannels > 1 & $imgSize > 1 ){
  for ( i in 2:$numChannels ){
    rowBuild = rbind(rowBuild, seq(i,$numChannels*$imgSize,$numChannels))
  }
}

colBuild = rowBuild
if ( $numImg > 1 ){
  for ( j in 2:$numImg ){
    rowBuild = seq((j-1)*$imgSize*$numChannels+1,j*$imgSize*$numChannels,$numChannels)
    if ( $numChannels > 1 ){
      for ( i in 2:$numChannels ){
        rowBuild = rbind(rowBuild, seq((j-1)*$imgSize*$numChannels+i,j*$numChannels*$imgSize,$numChannels))
      }
    }
    colBuild = cbind(colBuild, rowBuild)
  }
}

# Set input variables
x = t(colBuild)
w=matrix(1,rows=$numFilters, cols=$numChannels*$filterSize)

output = conv1d::backward_data(w, x, $pad, $stride, $numImg, $numChannels, $imgSize, $numFilters, $filterSize)

#Write output
write(output, $output)