Sha256: d24b63e6bc0c797a0a17c569e9a3c632121a174a33e39d222a6159ddb6b0e5a7
Contents?: true
Size: 709 Bytes
Versions: 54
Compression:
Stored size: 709 Bytes
Contents
module Torch
module NN
class Conv1d < ConvNd
def initialize(in_channels, out_channels, kernel_size, stride: 1,
padding: 0, dilation: 1, groups: 1, bias: true, padding_mode: "zeros")
kernel_size = _single(kernel_size)
stride = _single(stride)
padding = _single(padding)
dilation = _single(dilation)
super(in_channels, out_channels, kernel_size, stride, padding, dilation, false, _single(0), groups, bias, padding_mode)
end
def forward(input)
if @padding_mode == "circular"
raise NotImplementedError
end
F.conv1d(input, @weight, @bias, @stride, @padding, @dilation, @groups)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems