Class: Chainer::Functions::Array::Reshape

Inherits:

Chainer::FunctionNode

• Object
• Chainer::FunctionNode
• Chainer::Functions::Array::Reshape

Defined in:

lib/chainer/functions/array/reshape.rb

Overview

Reshapes an input array without copy.

Instance Attribute Summary

Attributes inherited from Chainer::FunctionNode

#inputs, #outputs, #rank

Class Method Summary

• .reshape(x, shape) ⇒ Object

Instance Method Summary

• #backward(indexes, grad_outputs) ⇒ Object
• #forward(inputs) ⇒ Object
• #initialize(shape) ⇒ Reshape constructor

  A new instance of Reshape.

Methods inherited from Chainer::FunctionNode

#apply, #backward_accumulate, #forward_cpu, #get_retained_inputs, #get_retained_outputs, #label, #output_data, #retain_inputs, #retain_outputs, #unchain

Constructor Details

#initialize(shape) ⇒ Reshape

Returns a new instance of Reshape.

# File 'lib/chainer/functions/array/reshape.rb', line 6

def initialize(shape)
  @shape = shape
end

Class Method Details

.reshape(x, shape) ⇒ Object

# File 'lib/chainer/functions/array/reshape.rb', line 10

def self.reshape(x, shape)
  return Chainer::Variable.as_variable(x) if x.shape == shape
  return self.new(shape).apply([x]).first
end

Instance Method Details

#backward(indexes, grad_outputs) ⇒ Object

# File 'lib/chainer/functions/array/reshape.rb', line 21

def backward(indexes, grad_outputs)
  gx = grad_outputs.first
  [Reshape.reshape(gx, @inputs.first.shape)]
end

#forward(inputs) ⇒ Object

# File 'lib/chainer/functions/array/reshape.rb', line 15

def forward(inputs)
  x = inputs.first
  new_shape = @shape.map { |s| s == -1 ? nil : s }
  [x.reshape(*new_shape)]
end