Sha256: 44aad5426f1c164e8eb4f4459c66ee9d72da91d8f7b1e473fa67817d6f26395f
Contents?: true
Size: 305 Bytes
Versions: 9
Compression:
Stored size: 305 Bytes
Contents
TensorStream::OpMaker.define_operation :cos do |op|
op.what_it_does "Computes cos of input element-wise."
op.parameter :input_a, "tensor X", validate: 'FLOATING_POINT_TYPES'
op.option :name, "Optional name", :nil
op.define_gradient do |grad, node, params|
-grad * ts.sin(params[0])
end
end
Version data entries
9 entries across 9 versions & 1 rubygems