Function Square

tensorflow/go/op/wrappers.go:32164–32176

Computes square of x element-wise. I.e., \\(y = x * x = x^2\\).

(scope *Scope, x tf.Output)

//
// I.e., \\(y = x * x = x^2\\).
func Square(scope *Scope, x tf.Output) (y tf.Output) {
	if scope.Err() != nil {
		return
	}
	opspec := tf.OpSpec{
		Type: "Square",
		Input: []tf.Input{
			x,
		},
	}
	op := scope.AddOperation(opspec)
	return op.Output(0)
}

// Returns which elements of x are NaN.
//