Sha256: 222f2339015a23c070f1d5623cb3fd69a48150b40ad4ea73f33e9cc74c3c740e
Contents?: true
Size: 578 Bytes
Versions: 9
Compression:
Stored size: 578 Bytes
Contents
TensorStream::OpMaker.define_operation :argmin do |op|
op.what_it_does "Returns the index with the smallest value across axes of a tensor."
op.parameter :input_a, "tensor X", validate: 'NUMERIC_TYPES'
op.parameter :axis, "Describes which axis of the input tensor to reduce across. For vectors, use axis = 0", :nil, validate: 'INTEGER_TYPES'
op.option :name, "Optional name", :nil
op.option :dimension, "Same as axis", :nil
op.option :output_type, "Output data type defaults to int32", ":int32"
op.define_gradient do |grad, node, params|
[nil, nil]
end
end
Version data entries
9 entries across 9 versions & 1 rubygems